恒源云(Gpushare)_自动化训练小技巧白送给你,不要吗?

文章来源 | 恒源云社区

原文地址 | 自动化后台训练

作者 | Littleor


大噶好啊,今日小编还在坚挺中,没疯,虽然被告知所在区域是封控区,但工作还需要我去完成,我得加油!⛽️

今天先不搬运小伙伴们的论文类文章,先给各位来个技巧分享!
内容不长,希望能带给各位炼丹小伙伴们一点小小的帮助。
恒源云(Gpushare)_自动化训练小技巧白送给你,不要吗?_第1张图片

这里是一个简单易用的一个后台全自动化训练日志记录、结果数据自动保存和自动关机的脚本。
仓库地址: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

你可能感兴趣的:(sshoss)