- 博客(3)
- 收藏
- 关注
原创 pytorch 预训练模型 最后几层的修改方法
pytorch 预训练模型 最后几层的修改方法方法1 model = net() #自己定义的模型,但要保证前面保存的层和自定义的模型中的层一致 state_dict = torch.load('xxx/xxx.pth')# 模型 keys = [] for k,v in state_dict.items(): if k.startswith('_fc'): continue
2021-01-05 12:36:05
2822
原创 Anaconda使用conda网络错误CondaHTTPError: HTTP 000 CONNECTION FAILED for url
Anaconda使用conda网络错误CondaHTTPError: HTTP 000 CONNECTION FAILED for url在创建新的虚拟环境时conda create -n pytorch1.5出现查看了博客的方法:http://www.manongjc.com/detail/15-rwhublzgdjutlfu.html首先在命令行中输入conda config --add channels - https://mirrors.tuna.tsinghua.edu.cn/an
2020-06-09 23:14:37
589
原创 Ubuntu 安装pytorch指定版本
Ubuntu 安装pytorch指定版本步骤1为了不与原有的pytorch版本冲突,建立conda 的虚拟环境在命令行中输入:conda -n pytorch1.2source activate pytorch1.2则命令行中会显示(pytorch1.2)步骤2进入pytorch官网pytorch官网链接若不是选择最新版本,可进入链接查找以前版本此时可以采用换源下载,提高下载速度本人采用了此博客的方法:https://blog.csdn.net/watermelon1123
2020-06-03 23:17:49
2317
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人