Skip to content

Bug Fixed and Adjustment on fedavg_v1.2.0#50

Open
AlexEisie wants to merge 1 commit intoSMILELab-FL:masterfrom
AlexEisie:master
Open

Bug Fixed and Adjustment on fedavg_v1.2.0#50
AlexEisie wants to merge 1 commit intoSMILELab-FL:masterfrom
AlexEisie:master

Conversation

@AlexEisie
Copy link

  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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant