当前位置:网站首页>pytorch训练进程被中断了
pytorch训练进程被中断了
2022-07-05 11:13:00 【IMQYT】
吓死我了,训练3天的模型的进程被自己手欠杀了,差点就哭了,租了一星期的服务器的钱打水漂了吗,时间都白费了吗,还能不能补救啊!第一次遇到这种情况,而且自己的代码运行很慢了(RTXA5000,按理说不慢才对,数据量太大了,为了减少日志的IO浪费的时间,就没有写日志),只保存了模型。已经手在抖了
话不多说,如何补救呢?
代码里保存模型只用了torch.save.其他参数都没有保存。epoch什么的都没保存,找了大量的经验,终于找到补救方法
重新加载模型
path='autodl-tmp/GraphDTA-master/model_GINConvNet_kiba.model'
model.load_state_dict(torch.load(path))
这样的话,模型学习到的内容都回来了,包括loss之类的。
从这里我看到,loss确实继续了294次的训练,预测值也是一样继续294次之后的结果,还好捡回来了,但是遇到一个问题,因为我看到epoch似乎重新从1开始了,这样的话不就还要训练600次?,所以记得修改下epoch的总次数,600-294=306,虽然控制中断写这1,但再训练306次就会结束。大功告成
边栏推荐
- Characteristics and electrical parameters of DDR4
- DDR4的特性与电气参数
- Huawei equipment configures channel switching services without interruption
- [advertising system] incremental training & feature access / feature elimination
- Oneforall installation and use
- Technology sharing | common interface protocol analysis
- 7.2每日学习4
- 关于vray 5.2的使用(自研笔记)(二)
- Msfconsole command encyclopedia and instructions
- Honing · fusion | know that the official website of Chuangyu mobile terminal is newly launched, and start the journey of digital security!
猜你喜欢
Some understandings of heterogeneous graphs in DGL and the usage of heterogeneous graph convolution heterographconv
不要再说微服务可以解决一切问题了!
修复动漫1K变8K
7.2每日学习4
COMSOL--三维图形的建立
2022 chemical automation control instrument examination questions and online simulation examination
9、 Disk management
关于vray5.2怎么关闭日志窗口
关于vray 5.2的使用(自研笔记)
关于vray 5.2的使用(自研笔记)(二)
随机推荐
九、磁盘管理
2022 Pengcheng cup Web
How to introduce devsecops into enterprises?
Deepfake tutorial
When using gbase 8C database, an error is reported: 80000502, cluster:%s is busy. What's going on?
Pull up loading principle
关于 “原型” 的那些事你真的理解了吗?【上篇】
Basic testing process of CSDN Software Testing Introduction
DDoS attack principle, the phenomenon of being attacked by DDoS
Explanation of message passing in DGL
AUTOCAD——遮罩命令、如何使用CAD对图纸进行局部放大
DOM//
2022 chemical automation control instrument examination questions and online simulation examination
Wechat nucleic acid detection appointment applet system graduation design completion (6) opening defense ppt
comsol--三维图形随便画----回转
OneForAll安装使用
Dspic33ep clock initialization program
Summary of websites of app stores / APP markets
A mining of edu certificate station
解决访问国外公共静态资源速度慢的问题