Testing - Model checkpoint to load for tieredImageNet

csyanbin / TPN-pytorch

Pytorch Code for ICLR19 paper: Learning to Propagate Labels: Transductive Propagation Network for Few-shot Learning.

175 stars 29 forks source link

Testing - Model checkpoint to load for tieredImageNet #7

Open oveilleux opened 3 years ago

oveilleux commented 3 years ago

Hi, first of all congrats on your great work and thank you for sharing.

I'm trying to reproduce your results on tiered-ImageNet. I already trained the model. When running the testing code, which "iters" value should I load the trained model from to obtain your results displayed in your paper? That parameter is indicated for the testing phase on mini-Imagenet, but it is not for tiered-ImageNet.

Thank you for your answer.

csyanbin commented 3 years ago

In our paper, we monitor the best validation performance and choose the corresponding iteration to test the model. This iteration may differ according to different training processes with random seeds.