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

相关阅读:
162. 寻找峰值
kubernetes集群编排——k8s存储
SpringBoot学习笔记（三）自动装配
MongoDB的安装及命令行操作
在服务器上开Juypter Lab教程（远程访问）
Web自动化测试流程：从入门到精通，帮你成为测试专家！
前端请求后台接口失败处理逻辑
DPDK原理探索: igb_uio
Vue3+TS+Vite 搭建组件库记录
Meta Llama 3本地部署

原文地址：https://blog.csdn.net/weixin_43717681/article/details/126644262