
PyTorch学习之路
一位去南理的学长推荐的一本书,叫做《Python深度学习基于PyTorch》。本专栏旨在记录在研读这本书的过程中,这本书没有提到的一些坑。
氩氪氙氡_
环保,海洋,机器人
展开
-
PyTorch实现CIFAR-10分类代码
这篇是PyTorch学习之路第七篇,用于记录PyTorch实现CIFAR-10分类代码(书上的代码有好多冗余)目录完整代码(还未训练)完整代码(已训练,直接载入模型)完整代码(还未训练)import torchimport torchvisionimport torchvision.transforms as transformsfrom torch.utils.data import DataLoadertrain_batch_size = 4test_batch_size = 4n原创 2021-01-25 10:25:06 · 2718 阅读 · 9 评论 -
len(train_loader),len(label)和训练数据集个数之间的关系
len(train_loader),len(label)和训练数据集之间的关系。train_loader中的每个元素相当于一个分组,一个组中有4张图片,label就是一个分组中的一张图片的标签原创 2021-01-25 08:57:19 · 15336 阅读 · 2 评论 -
PyTorch如何加载已经训练好的网络模型
PyTorch要加载已经训练好的网络模型,需要保留什么代码,增加什么代码?model.load_state_dict(torch.load('./model/model.pth'))#再加载网络的参数model = model.to(device)原创 2021-01-24 22:49:12 · 12758 阅读 · 1 评论 -
python写路径时\和/的区别是什么
本篇是Pytorch学习之路的第四篇问题搞混了python写路径时\和/的区别解答由此可见,E:/pythonMixture/test2.txt和E:\\pythonMixture\\test2.txt没区别。参考链接python保存路径/与\的区别(转载)原创 2021-01-24 22:24:24 · 5729 阅读 · 0 评论 -
如何保存pytorch 训练好的网络模型
Pytorch要保存训练好的模型,但是报错:PermissionError: [Errno 13] Permission denied。通过在相对路径的最后加上保存模型的文件名解决问题原创 2021-01-24 22:12:20 · 1409 阅读 · 0 评论 -
Pytorch从本地获取CIFAR数据集报错
问题描述为加载本地CIFAR文件,我解压后放在了data文件夹下(data文件夹里有其他文件了)然后:按照书中的把download设为False,会报错;把download设为True,会报错解决方案download设为True新建一个文件夹(不妨命名dataset),把CIFAR10数据集放入其中成果图原因分析可能是因为之前的data文件夹下有其他数据集了...原创 2021-01-22 11:35:42 · 383 阅读 · 0 评论 -
TensorFlow installation not found - running with reduced feature set.
按照教程,希望打开网页时,命令行中出现TensorFlow installation not found - running with reduced feature set. 解决方案:cmd中用相对路径而非绝对路径原创 2021-01-19 11:50:39 · 27634 阅读 · 4 评论