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