没有合适的资源?快使用搜索试试~ 我知道了~
pytorch-mdn:PyTorch的混合物密度网络
共15个文件
py:7个
png:2个
makefile:1个
需积分: 50 4 下载量 135 浏览量
2021-04-03
13:18:50
上传
评论
收藏 305KB ZIP 举报
温馨提示
pytorch-mdn 此回购包含的代码。 用法: import torch . nn as nn import torch . optim as optim import mdn # initialize the model model = nn . Sequential ( nn . Linear ( 5 , 6 ), nn . Tanh (), mdn . MDN ( 6 , 7 , 20 ) ) optimizer = optim . Adam ( model . parameters ()) # train the model for minibatch , labels in train_set : model . zero_grad () pi , sigma , mu = model ( minibatch ) loss
资源推荐
资源详情
资源评论
收起资源包目录
pytorch-mdn-master.zip (15个子文件)
pytorch-mdn-master
setup.py 792B
.gitignore 39B
requirements.txt 27B
Makefile 80B
data
before.png 149KB
after.png 161KB
.travis.yml 102B
LICENSE 1KB
README.md 1KB
tests
__init__.py 0B
basic_test.py 722B
gradient_test.py 811B
example
example.py 2KB
mdn
__init__.py 0B
mdn.py 4KB
共 15 条
- 1
资源评论
地下蝉
- 粉丝: 30
- 资源: 4527
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功