深度学习模型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)