- 博客(4)
- 收藏
- 关注
原创 踩坑记录:Pytorch微调resnet只保存输出层参数,为什么重新加载会效果奇差!
最近在用Pytorch微调resnet50和resnet101模型。因为只微调了输出层秉着节约的原则,没有保存完整的模型,我每5个epoch保存一次输出层的参数(weight和bias),并且训练了100个epoch,看看损失函数,收敛了。然后开始用test测试,看看混淆矩阵怎么样加载预训练模型,并把保存的输出层参数加载到我们的模型上面,看看结果model_ft = models.resnet50(pretrained=True) num_fc_ftr = model_ft.fc.in_fe
2020-05-13 04:11:52 1136
原创 深度学习中使用shell脚本实现多进程并发/挂机炼丹/学术实验
前情概要:拥有多块GPU,需要跑多个模型测试baseline所以需要同时跑多个训练代码,使用shell脚本和python的argparse库解析参数是最方便的。1.argparse解析参数现在看下可选参数的用法,所谓可选参数,也就是命令行参数是可选的# -*- coding: utf-8 -*-import argparseparser = argparse.ArgumentPar...
2020-05-01 14:07:28 2941
原创 nohup和&的使用/2>&1是什么意思/怎么关闭nohup挂起的程序
1.&和nohup和2>&1的定义:& 放在命令到结尾,表示后台运行,防止终端一直被某个进程占用,这样终端可以执行别到任务nohup放在命令的开头,表示不挂起(no hang up),也即,关闭终端或者退出某个账号,进程也继续保持运行状态0 表示stdin标准输入1 表示stdout标准输出2 表示stderr标准错误2>&1 也就表示将错误...
2020-05-01 13:55:36 1411
原创 问题 I: Congestion Charging Zone
题目描述Tehran municipality has set up a new charging method for the Congestion Charging Zone (CCZ) which controls the passage of vehicles in Tehran’s high-congestion areas in the congestion period (CP) ...
2019-04-06 19:49:06 239
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人