From c7a3d34d880043576a1c4d04926d857bfb1e5f72 Mon Sep 17 00:00:00 2001 From: Yunhao Meng Date: Thu, 23 Oct 2025 13:07:55 +0800 Subject: [PATCH] =?UTF-8?q?=E9=85=8D=E7=BD=AE=E6=9B=B4=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- config/coco_cfg.yaml | 12 ++++++------ utils/args.py | 4 ++-- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/config/coco_cfg.yaml b/config/coco_cfg.yaml index 34175e9..6faa286 100644 --- a/config/coco_cfg.yaml +++ b/config/coco_cfg.yaml @@ -4,21 +4,21 @@ model_name: "yolo_v11_n" # yolo_v11_n, yolo_v11_t, yolo_v11_s, yolo_v11 i_seed: 202509 # initial random seed num_client: 64 # total number of clients -num_round: 5 # total number of communication rounds +num_round: 50 # total number of communication rounds num_local_class: 80 # number of classes per client res_root: "results" # root directory for results -dataset_path: "/home/image1325/ssd1/dataset/COCO128/" +dataset_path: "/mnt/DATA/coco" # root directory for dataset # train_txt: "train.txt" # path to training set txt file # val_txt: "val.txt" # path to validation set txt file # test_txt: "test.txt" # path to test set txt file local_batch_size: 32 # local training batch size -val_batch_size: 4 # validation batch size +val_batch_size: 128 # validation batch size -num_workers: 4 # number of data loader workers -min_data: 128 # minimum number of images per client -max_data: 128 # maximum number of images per client +num_workers: 8 # number of data loader workers +min_data: 1700 # minimum number of images per client +max_data: 1800 # maximum number of images per client partition_mode: "overlap" # "overlap" or "disjoint" connection_ratio: 1 # connection ratio, e.g., 1.0 means all clients diff --git a/utils/args.py b/utils/args.py index b965f7f..3696ab2 100644 --- a/utils/args.py +++ b/utils/args.py @@ -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()