报错:‘tuple’ object has no attribute ‘softmax’
可能是
return x, aux
应该写成
if self.training:
return x, aux
else:
return x
报错:‘tuple’ object has no attribute ‘softmax’
可能是
return x, aux
应该写成
if self.training:
return x, aux
else:
return x