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 `allow_unused` arg for models with task-specific parameters #145

Open rees-c opened 2 years ago

rees-c commented 2 years ago

Hi,

Thanks for building this library. I wanted to add the allow_unused argument when computing gradients to allow for meta-learners with task-specific parameters. I left the default behavior unchanged.