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

1. 对于train_data 使用batch_size的方式进行训练，一定要设置batch的类型，所以规定train_data一定要加语句transform=torchvision.transforms.ToTensor()

1 2 3 4 5 6

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'>