def forward(self, input):
if isinstance(input, Variable):
return torch.sigmoid(input)
elif isinstance(input, tuple) or isinstance(input, list):
return my_data_parallel(self, input)
else:
raise RuntimeError('unknown input type')
评论列表
文章目录