Pytorch编程心得(2018/10/27-2018/10/29)

  1. Numpy中不包含unsquueze函数,但是pytorch中是包含了这个函数的,也就是说unsqueeze函数可以作用在tensor上,但是不能作用在ndarray上。
  2. 需要计算梯度的Variable是不可以转化为numpy数组的,需要先试用detach方法或者data方法,然后再转化为Numpy数组。
  3. 如果出现下述问题:“forward() takes 1 positional argument but 2 were given”,那么很有可能是在定义forward函数的时候,没有加入self参数。
  4. torch.max()返回两个结果,第一个是最大值,第二个是对应的索引值;第二个参数 0 代表按列取最大值并返回对应的行索引值,1 代表按行取最大值并返回对应的列索引值。
  5. Pytorch的训练模式和测试模式切换
    由于训练的时候Dropout和BN层起作用,每个batch BN层的参数不一样,dropout在训练时随机失效点具有随机性,所以训练和测试要区分开来。使用时切记要根据实际情况切换:model.train()和model.eval()
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值