|
|
module 'torch' has no attribute 'softmax'
source_unseen_output = torch.softmax(torch.mm(source_fc8, unseen_vec.t()) * self.entropy_lambda, dim=1) E_loss = (-source_unseen_output * source_unseen_output.log()).sum(1).mean() loss2 = E_loss.item() CE_loss += args.entropy_loss_weight * E_loss
File "C:/Users/Dell/Downloadstrain.py", line 158, in train
source_unseen_output = torch.softmax(torch.mm(source_fc8, unseen_vec.t()) * self.entropy_lambda, dim=1)
AttributeError: module 'torch' has no attribute 'softmax'
0 提建议
分享
编辑
收藏
删除
结题
收藏
举报
追加酬金
(90%的用户在追加酬金后获得了解决方案)
当前问题酬金
¥ 0 (可追加 ¥500）
支付方式
扫码支付
加载中...
2 条回答 默认 最新
-
爱晚乏客游 2021-05-15 19:23关注
检查下torch的版本
本回答被题主选为最佳回答 , 对您是否有帮助呢? 本回答被专家选为最佳回答 , 对您是否有帮助呢? 本回答被题主和专家选为最佳回答 , 对您是否有帮助呢?举报
分享
按下Enter换行，Ctrl+Enter发表内容
