Skip to content

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