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

torch版本的原因,torch . . 使用softmax时需要加上维度参数: p torch.nn.functional.softmax x,dim ...

2018-11-13 16:53 0 1085 推荐指数:

查看详情

Can not squeeze dim[1], expected a dimension of 1

遇上如题所示问题,多半是因为使用了sparse_categorical_crossentropy作为损失函数 这会出现在分类问题中,这个激活函数要求如下: labels vect ...

Wed Feb 05 23:21:00 CST 2020 0 1053
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM