Parameter setting for the n-way k-shot learning

[tdawn]

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