Python data_parallel_modelの例

コード例 #1

 def forward(self, x):
     output = n_util.data_parallel_model(self.cnn, x, self.ngpu)
     return output.view(x.size(0), -1)

コード例 #2

ファイル: deconv_pow2.py プロジェクト: behroozsepehry/VAE

 def forward(self, x):
     input = x.view(x.size(0), -1, 1, 1)
     output = n_util.data_parallel_model(self.cnn, input, self.ngpu)
     return output

コード例 #3

 def forward(self, input):
     output = n_util.data_parallel_model(self.mlp, input, self.ngpu)
     return output