之前我曾经写过一次关于如何使用服务器进行神经网络模型训练的博客,有兴趣的可以翻出来看看
今天我想介绍的是源于我下面的一种的训练情况:
训练一个神经网络模型,需要2天2夜,我不仅仅需要最后一个模型的模型性能指标,还需要中间几个的模型训练参数,如果们使用tmux进行开放进程进行模型的训练。
这时候我们就需要使用screen指令将Linux的终端输出的日志进行收集生成一个文件
在进行模型训练之前,运行指令:
script screen.log
运行完上述指令后
表示日志文件已经开始收集终端的输出log
执行网络模型的运行指令
python3 train.py --img 416 --batch 40 --epochs 5 --data coco128.yaml --weights yolov5s.pt --device 0
在运行神经网络训练结束后,
运行指令:
exit
终端输出:
查看screen.log文件中包含了所有的终端输出日志
包含了所有的训练日志,十分方便!