serWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument.

torch版本的原因,torch0.4.1使用softmax时需要加上维度参数:

p=torch.nn.functional.softmax(x,dim=1)

猜你喜欢

转载自www.cnblogs.com/yangruicvpr/p/9953110.html