flennerhag / warpgrad

Meta-Learning with Warped Gradient Descent
https://openreview.net/forum?id=rkeiQlBFPB
Apache License 2.0
92 stars 18 forks source link

Parameter setting for the n-way k-shot learning #10

Open tdawn opened 3 years ago

tdawn commented 3 years ago

Hi Flennerhag,

I want to reproduce the accuracy of Warp-Leap in Table 1 for the Omniglot dataset. However, I am confused with the parameter setting to run n-way k-shot learning with this repository. In main.py of Omniglot implementation, it seems that you set the parameter for 20-way (from --classes input), but I do not see the argument to set the number of shots there.

Could you please show how to set the number of shots for this implementation, i.e., 20-way 100-shot or maybe another scenario for 5-way 5-shot?

Thanks, Dwi