search

learnables / learn2learn

A PyTorch Library for Meta-learning Research
http://learn2learn.net
MIT License
2.61k stars 350 forks source link

How to change the batchsize of a Metadata class instance #373

Closed woreom closed 1 year ago

woreom commented 1 year ago

I trying to use learn2learn/examples/vision/protonet_miniimagenet.py/ and the batchsize is fixed to 8, how can I change it?

seba-1511 commented 1 year ago

This implementation of ProtoNet uses a batch size of 1. See the MAML example for how to batch over tasks.