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

Fix Error: AttributeError: 'collections.OrderedDict' object has no attribute 'half'

Browse Source
This commit is contained in:
Yunhao Meng
2025-11-03 21:12:35 +08:00
parent fdb70869f9
commit f658c6aca5
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
fed_run.py
View File

@@ -14,6 +14,7 @@ from fed_algo_cs.server_base import FedYoloServer
from utils.args import args_parser # args parser from utils.args import args_parser # args parser
from utils.fed_util import divide_trainset # divide_trainset from utils.fed_util import divide_trainset # divide_trainset
from utils import util from utils import util
from utils import fed_util
from utils.fed_util import prepare_result_dir from utils.fed_util import prepare_result_dir
@@ -189,7 +190,9 @@ def fed_run():
# Save final global model weights # Save final global model weights
# FIXME: save model not adaptive YOLOv11-pt specific # FIXME: save model not adaptive YOLOv11-pt specific
save_model = {"config": cfg, "model": copy.deepcopy(global_state if global_state else None)} global_model = fed_util.init_model(model_name, num_classes=len(cfg["names"]))
global_model.load_state_dict(global_state)
save_model = {"config": cfg, "model": copy.deepcopy(global_model if global_model else None)}
torch.save(save_model, f"{weights_root}/last.pt") torch.save(save_model, f"{weights_root}/last.pt")
if best == mAP: if best == mAP:
torch.save(save_model, f"{weights_root}/best.pt") torch.save(save_model, f"{weights_root}/best.pt")