更新配置文件路径,优化文档字符串,增强代码可读性

This commit is contained in:
2025-10-31 13:14:29 +08:00
parent 194ca8ee31
commit c7afef2dc2
2 changed files with 42 additions and 10 deletions

View File

@@ -7,7 +7,7 @@ def args_parser():
parser.add_argument("--epochs", type=int, default=16, help="number of rounds of local training")
parser.add_argument("--input_size", type=int, default=640, help="image input size")
parser.add_argument("--config", type=str, default="./config/coco_cfg.yaml", help="Path to YAML config")
parser.add_argument("--config", type=str, default="./config/uav_cfg.yaml", help="Path to YAML config")
args = parser.parse_args()