Train
- copy一份之前已有的main和pre的cfg文件和commands文件到新建目录。0414目录是之前训练时候使用的目录,该目录下除main.cfg、pre.cfg、commands外的其他文件都是训练过程中产生的,因此不需要拷贝。
- 修改main.cfg和pre.cfg、commonds文件
1、修改data路径
2、修改train参数:
start_checkpoint -- 需要改成上次训练得到的最佳checkpoint文件。
train_dir -- 改成新创建的用于保存训练log和结果的目录 0608
对pre.cfg的修改如下:
这里的修改需要注意的是:如果start_checkpoint改成相对路径如(~/project/***)会出现找不到需要的文件的错误。
3、修改commonds文件
- 修改cmd2sh.py脚本,修改train.py的路径为该脚本在自己环境中的位置。还要修改自己的conda.sh的路径,激活训练环境需要的conda env。