学习笔记-爬出坑-大杂烩

更换路径

pytorch中封装了很多实现,需要用到官方下载的数据集,如果本地有下载好的就不需要再下载了
https://www.freesion.com/article/5198895969/
看这里,改路径然后就可以用本地的了
运行时出现
在这里插入图片描述

GPU的坑

GPU的环境没装好,所以如果选择GPU环境,模型跑不起来
测试语句:
在这里插入图片描述
两个都输出说明当前环境的cuda可以正常使用
否则:
只能用CPU环境,

带镜像安装库

pip install opencv==3.4.2 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

分而治之思想

模型训练时读取数据集格式,
可以先了解正确的读取格式
再写个脚本直接读,读正确了再放到模型中进行读取

猜你喜欢

转载自blog.csdn.net/weixin_42630613/article/details/107840250
今日推荐