深度学习模型命令行传参——断点调试解决方案
深度学习模型debug
问题
在深度学习中，经常见到训练代码如下所示，通过命令行进行参数传递，但是通过这种方法，不利于我们使用pycharm自带的调试debug程序。
解决方案
新建一个py文件，通过调用subprocess库，然后通过运行这个文件，进行命令行调用，从而达到代码断点调试的效果
import subprocess
train_file = "../data/cave/cave_train"
eval_file = "../data/cave/cave_val"
output_folder = "outputs"
# 构建命令行命令
command = ["python", "train.py",
"--train-file", train_file,
"--eval-file", eval_file,
"--outputs-dir", output_folder,
"--model", "PDcon_SSF"
]
# 执行命令
subprocess.call(command)