Shell 在训练模型的时候自动保存训练文件和模型到指定文件夹

在进行深度学习训练的过程中,往往会跑很多实验,这就导致有的实验设置会忘记或者记混淆,我们最好把train test model 的代码都copy一遍到指定文件夹中,这样后面检查也方便。
用shell指令保存文件

#!/bin/sh
GRUB_CMDLINE_LINUX="iommu=soft"
export CUDA_VISIBLE_DEVICES=1,2,3,4


save_dir=./saved_distill_128_fea_ctx
if [ ! -d $save_dir  ];then  #判断文件夹是否存在
  mkdir $save_dir
  echo mkdir $save_dir
else
  echo dir exist $save_dir
fi
nohup python -u train_bxh.py \
--train_data /search/odin/boqin/BIAN/scatter-2020_new/haoweilai-wiki-110w-train-lmdb \
--valid_data /search/odin/boqin/BIAN/scatter-2020_new/haoweilai-wiki-110w-train-lmdb \
--character ./dictionary_v1.txt \
--batch_size 45 \
--PAD \
--exp_name hwl-wiki-110w-exp-opencv-test \
--lr 0.5 \
--feature_rate 1 \
--ctx_rate 1 \
--lambda_kl 50 \
--output_channel_student 128 \
--hidden_size_student 128 \
--inter_rate 4 \
--print_interval 10 \
--original_model_path ./epoch_12.pth \
--save_path $save_dir \
1>$save_dir/log.txt  2>&1 & 

cp ./train_bxh.py $save_dir  # copy 文件到指定的文件夹中
cp ./model.py $save_dir
cp ./train.sh $save_dir
cp -R ./modules $save_dir

# --check_point /home/qinbo/BIAN/scatter-2020_new/saved_model_distill_7/hwl-wiki-110w-exp-opencv-test/epoch_2_iteraiton_20500.pth \
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值