文章来源 | 恒源云社区
原文地址 | 自动化后台训练
作者 | Littleor
大噶好啊,今日小编还在坚挺中,没疯,虽然被告知所在区域是封控区,但工作还需要我去完成,我得加油!⛽️
今天先不搬运小伙伴们的论文类文章,先给各位来个技巧分享!
内容不长,希望能带给各位炼丹小伙伴们一点小小的帮助。
这里是一个简单易用的一个后台全自动化训练日志记录、结果数据自动保存和自动关机的脚本。
仓库地址:https://github.com/Littleor/HY-Auto-Trainer
特性
- 使用
nohup
在后台进行训练,完全可以断开ssh
的连接,解放双手。 - 训练完成自动打包运行结果到
OSS
上,避免结果丢失。 - 上传完成自动关机,省钱省事~
用法
配置
- 复制目录下所有的 *.sh 文件到项目目录(如 /hy-tmp)
给 *.sh 文件加上权限:
chmod a+x *.sh
- 修改 run.sh 中的 logPath 来修改日志保存路径
- 修改 end.sh 中的 resultPath 来修改需要保存的运行结果的路径
- 如果非 Python 项目, 则需要修改 end.sh 的 python 为对应的执行命令即可
使用
以本项目为例,在项目中有个 Train/main.py 文件作为模拟训练的文件,并将项目文件放在 /hy-tmp 目录下。
运行方法:
# ./run.sh [可执行文件目录] [Tag 用于标注结果]
./run.sh /hy-tmp/Train/main.py Test-Trainer