pytorch运行程序过程中遇到的问题

将pytorch更新到0.4.0最新版后对0.3.1版本代码会有如下警告,它在提醒用户下个版本这将成为一个错误

错误1:UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use tensor.item() to convert a 0-dim tensor to a Python number  train_loss += loss.data[0]

修改如下:

#原语句:
train_loss+=loss.data[0]
#修改后:
train_loss+=loss.item()

错误2:UserWarning: volatile was removed and now has no effect. Use `with torch.no_grad():` instead.
v = Variable(torch.from_numpy(x).type(dtype), requires_grad = False, volatile = True)

修改如下:

#原语句
v = Variable(torch.from_numpy(x).type(dtype), requires_grad = False, volatile = True)
#修改后语句
v = Variable(torch.from_numpy(x).type(dtype), requires_grad = False)

错误3:runtimewarning:invalid value encountered in divide

修改如下:

import numpy as np 
np.seterr(divide='ignore', invalid='ignore')

错误4:Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X

修改如下:

#原语句
density_cls_prob = F.softmax(density_cls_score)
#修改后
density_cls_prob = F.softmax(density_cls_score,dim=1)

错误5:raise ImportError( "unittest2 is required to run tests with Python 2")

解决方法:

conda install unittest2

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值