【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 |
备注：
flag类型的参数为store_true，只需在命令行中出现即可设为True。- 若通过
--max_epoch或--max_steps手动指定训练规模，将覆盖 YAML 配置文件中对应字段。