如何从上一次训练断开处重新训练?
YoungP2001 opened this issue · 4 comments
YoungP2001 commented
您好,小白刚接触深度学习这方面,之前训练老是断开,还没有完成过一次。想问您怎么从上一次断开处重新训练?
我看见option的path里有
pretrain_model_G: ~
strict_load: true
resume_state: ~ # true
还有断开时的迭代次数和epoch怎么得到,我该怎么改代码?
Algolzw commented
你好,可以在log里你的project下面的training_state目录下面找到最近的节点,然后option中设置
resume_state: log/project_name/training_state/xxxx.state
即可以断点重训。
15049114601 commented
Algolzw commented
单纯断点重训的话pretrain_model_G这个不用管,它会从training_state里面自动加载对应的模型。strict_load是指要求加载的模型参数要和当前代码中的模型参数保持一致,一般设置为true就好。
15049114601 commented
单纯断点重训的话pretrain_model_G这个不用管,它会从training_state里面自动加载对应的模型。strict_load是指要求加载的模型参数要和当前代码中的模型参数保持一致,一般设置为true就好。
明白了,感谢!