search

tristandeleu / pytorch-meta

A collection of extensions and data-loaders for few-shot learning & meta-learning in PyTorch
https://tristandeleu.github.io/pytorch-meta/
MIT License
1.98k stars 256 forks source link

Add MetaModule._enable_meta_learning #149

Open pbsds opened 2 years ago

pbsds commented 2 years ago

This is a way to "disable" a MetaModule. This reduces complexity downstream for projects where meta-learning per module is conditional.

Otherwise I have to make both a non-meta and a meta-learned variant of each module.