今天和休息,有个学生和我打电话,抱怨最近总加班训练模型以及调优;
倍感诧异,训练模型为什么还要加班?
首先说明一下其实模型训练是可以后台执行的,不用你天天看着,接下来说一下流程:
- 在你的模型中务必添加模型保存代码,否则很有可能你高高兴兴回家了,模型训练的也狠OK,但是第二天发现模型没保存,然后回家拍大腿;
参考代码:
model.save_model("./ag_news_model_1")
- 执行nohup代码,并设置log日志,模型万一在哪个环节报错了,你好及时发现错误以及查看整个训练过程;
nohup python restudy_news.py >> log.txt &
# 按任意键退出
- 确认模型训练的进程是否已经开启;
ps -af | grep python
- 确认log日志是否设置成功;
- 查看实时log信息
# 实时查看log信息
tailf ./log/finbert_100.txt
# 停止查看log信息
control + c