文件名称：pytorch-mdn:PyTorch的混合物密度网络
文件大小：305KB
文件格式：ZIP
更新时间：2024-04-23 07:05:30
Python
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
----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)