Imported from https://github.com/crashmoon/MAML-Pytorch-Multi-GPUs