FedLab-benchmarks icon indicating copy to clipboard operation
FedLab-benchmarks copied to clipboard
verified publisher icon SMILELab-FL

Bug Fixed and Adjustment on fedavg_v1.2.0

Open AlexEisie opened this issue 10 months ago • 0 comments

  1. args.cuda is only be refered on model, but not be refered in params of SubsetSerialTrainer, which has a default value "False". This caused that pytorch wont train on gpus even args.cuda is setted to "True".
  2. data_indices can be replaced by Partitioner in fedlab.

AlexEisie avatar Mar 14 '25 17:03 AlexEisie