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()

否则就会出现错误：