Anchor-x/fed-yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

配置更改

Browse Source
This commit is contained in:
Yunhao Meng
2025-10-23 13:07:55 +08:00
parent b81f33ad28
commit c7a3d34d88
2 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
utils/args.py
View File

@@ -5,9 +5,9 @@ import os
def args_parser():
parser = argparse.ArgumentParser()
parser.add_argument("--epochs", type=int, default=10, help="number of rounds of local training")
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/uav_cfg.yaml", help="Path to YAML config")
parser.add_argument("--config", type=str, default="./config/coco_cfg.yaml", help="Path to YAML config")
args = parser.parse_args()