nupurkmr9 / S2M2_fewshot

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Cannot reproduce S2M2_R

voa18105 opened this issue · comments

Good day!
Is there a description how to reproduce your state-of-art results? I've tried to repeat your result with 1-shot S2M2_R on CUB, and got only 71%, should be 10% higher...
I could describe what exactly did I do, but may be it would be more useful for me and everyone if you could describe what did you do?
Launch params, choosing epoch or smth else, do not know, but 10% lower means smth is wrong with my reproduce.

Thank you in advance!

Hi,
We first train a model with rotation self-sueprvision with the following hyper-parameters for CUB dataset:
python train.py --dataset CUB --method rotation --model WideResNet28_10 --stop_epoch 200 --batch_size 64 --test_batch_size 16

We then finetune the model using both manifold mixup and rotation self-supervision using the command:
python train.py --dataset CUB --method S2M2_R --model WideResNet28_10 --stop_epoch 100 --batch_size 64 --test_batch_size 16

Let us know if you still face any issues.
Thanks.

Thank you for instruction! I will follow it