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