issues
search
masa-su
/
pixyz
A library for developing deep generative models in a more concise, intuitive and extendable way
https://pixyz.io
MIT License
491
stars
41
forks
source link
Feature/pytorch module
#100
Closed
ktaaaki
closed
4 years ago
ktaaaki
commented
4 years ago
torch.nn.DataParallelに対応するため,LossAPIの基礎部分を含めた変更を行いました.
変更点
Lossの親クラスをnn.Moduleに設定
Loss.train, Loss.testをそれぞれLoss.loss_train, Loss.loss_testにリネーム
初期化におけるsuper().
init
()の呼び出し順序を変更
DataParalleledLossクラスを追加
torch.nn.DataParallelに対応するため,LossAPIの基礎部分を含めた変更を行いました.
変更点