在PyTorch学习过程中所遇问题

1、RuntimeError: stack expects each tensor to be equal size, but got [3, 256, 256] at entry 0 and [3, 249, 256] at entry 26

报错原因:(1)没有转换图片格式,不同的图片格式的颜色空间不同

设置in_channel的值与读入图像img1的实际通道数不符合。

2、jupyter Notebook无法创建新文件,提示文件不可信,终端里显示jupyter后台自动退出:Notebook Untitled.ipynb is not trusted jupyter。

原因:系统用户名为中文导致,不能简单更改系统用户名,否则会导致系统奔溃

解决方法:在终端运行:pip install pyzmq==19.0.2

3、

在PyTorch学习过程中所遇问题_第1张图片

4、Module.cuda() missing 1 required positional argument: 'self'

5RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same or input should be a MKLDNN tensor and weight is a dense tensor

问题:输入类型是tensor类型,与权重类型(cuda)不符
 

将tensor转成numpy的几种情况:

  1. GPU的Variable变量:a.cuda().data.cpu().numpy()
  2. GPU中的tensor变量:a.cuda().cpu().numpy()
  3. CPU中的Variable变量:a.data.numpy()
  4. CPU中的tensor变量:a.numpy()

.cuda()是读取GPU中的数据

.data是读取Variable中的tensor

.cpu是把数据转移到cpu上

.numpy()是把tensor变成numpy

你可能感兴趣的:(pytorch,学习,numpy)