2018年11月13日
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) 阅读全文
posted @ 2018-11-13 16:53 勿忘初心233 阅读(2501) 评论(0) 推荐(0) 编辑