jupyter 挂掉的内核和transform=torchvision.transforms.ToTensor()
今日小问题
今天执行以下代码的时候，jupyter的内核不停挂掉。
plt.imshow(image.squeeze(),cmap='gray')
print('label:',label)
后经过查询
https://blog.csdn.net/Ky_11111/article/details/120635990
只需在代码前添加以下代码即可
import os
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"今日还有个小问题。
执行以上两个代码的时候一直报错，后经过学习。
1、TypeError:default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists; found
- 对于train_data 使用batch_size的方式进行训练，一定要设置batch的类型，所以规定train_data一定要加语句transform=torchvision.transforms.ToTensor()
| 1 | train_set=torchvision.datasets.FashionMNIST( root='./Data/FashionMNIST', train=True, transform=torchvision.transforms.ToTensor() # download=transforms.Compose([ # transforms.ToTensor() # ]#我们定义了应该在我们数据元素上执行的转换，组合类允许我们创建转换的组合， #在这种情况下我们只是把我们的数据变成了一个张量，这是一个单一的变换 ) |
否则就会出现错误：
| 1 | TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists; found <class 'PIL.Image.Image'> |