AntreasAntoniou / HowToTrainYourMAMLPytorch

The original code for the paper "How to train your MAML" along with a replication of the original "Model Agnostic Meta Learning" (MAML) paper in Pytorch.
https://arxiv.org/abs/1810.09502
Other
759 stars 137 forks source link

How to use dataparallel function of Pytorch in your implementation? #17

Open rshaojimmy opened 5 years ago

rshaojimmy commented 5 years ago

May I ask how to apply the dataparallel function of Pytorch on the network in your implementation? Faster training and larger batchsize can be achieved if parallelizing the MAML computational graph with dataparallel function. This currently seems not easy to do with Pytorch. I am very appreciated if you have any idea on this point.