【BBDM】main.py -- notes

命令行接口

python main.py [OPTIONS]

参数

参数	类型	默认值	说明
-c, --config	str	"BBDM_base.yml"	配置文件路径
-s, --seed	int	1234	随机种子，用于结果复现
-r, --result_path	str	"results"	结果保存目录
-t, --train	flag	False	训练模式开关：开启则执行训练，否则执行测试
--sample_at_start	flag	False	是否在程序启动即进行一次采样（主要用于调试）
--sample_to_eval	flag	False	是否在测试阶段额外进行采样用于评估
--save_top	flag	False	仅保存验证集上性能最好的模型检查点
--gpu_ids	str	"0"	指定可见 GPU 列表（逗号分隔），设为 "-1" 则全程使用CPU
--port	str	"12355"	DDP（分布式数据并行）通信端口
--resume_model	str	None	继续训练时加载的模型检查点文件路径
--resume_optim	str	None	继续训练时加载的优化器状态文件路径
--max_epoch	int	None	最大训练轮数，若指定则覆盖配置文件内的 n_epochs
--max_steps	int	None	最大训练步数，若指定则覆盖配置文件内的 n_steps

备注：

1. flag 类型的参数为 store_true，只需在命令行中出现即可设为 True。
2. 若通过 --max_epoch 或 --max_steps 手动指定训练规模，将覆盖 YAML 配置文件中对应字段。