[P] Memory Efficient Model-Agnostic Meta-Learning
PyTorch implementation of Model-Agnostic Meta-Learning with gradient checkpointing. This allows performing way (~10-100x) more MAML steps with the same GPU memory budget.
Code: https://github.com/dbaranchuk/memory-efficient-maml
Example: Open In Colab
Install: pip install torch_maml
submitted by /u/dbaranchuk
[link] [comments]