- 博客(3)
- 收藏
- 关注
RSS订阅原创 使用torchvision加载数据集mnist出错出错的解决方案
使用torchvision加载数据集mnist出错出错的解决方案错误提示为：RuntimeError: Dataset not found. You can use download=True to download it加载数据集代码为：trainset_mnist = torchvision.datasets.MNIST(root='d:/dataset/MNIST', train=True, download=False, transform=transform_train)错误的原因是’d:/
2020-12-18 20:07:19
1504
2
原创 2020-11-11
torch.tensor属性:grad 张量梯度grad_fn 梯度函数data 张量数据is_leaf 是否叶子节点devices 张量所在的设备（CPU/GPU）shape 张量的形状（64,3,244,244）dtype data的数据类型requires_grad 是否需要梯度
2020-11-11 20:08:57
43
原创 跑代码记录（backgrounds_challenge）
这里写自定义目录标题跑代码记录（backgrounds_challenge）数据准备model选择预测输出跑代码记录（backgrounds_challenge）数据准备利用mask合成验证集图片。设置mask所有值为1（表示保留）或者0（表示去除），这样与图片矩阵相乘即可获得想要的图片内容。# tensor的操作参数就是其shape,inp为背景图片，im为前景图片,tile()按指定shape铺开（复制）矩阵，图片类型为'uint8'。fg_mask = np.tile(fg_mask[:,
2020-10-30 22:08:22
334
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人