google-research / big_transfer

Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
https://arxiv.org/abs/1912.11370
Apache License 2.0
1.5k stars 174 forks source link

Top-1 accuracy on few-shot tasks #54

Closed fengliu90 closed 3 years ago

fengliu90 commented 3 years ago

Hi all,

I am reproducing results on few-shot tasks (cifar10). After running train.py (by setting examples_per_class as 5), I find that the accuracy reported in this repo (see few-shot section) and paper is actually the top-5 accuracy. Am I correct?


Sorry for the confusing comments. I changed get_resolution when I ran the experiment, which is the reason why my previous runs cannot get good top-1 accuracy. When I recover the get_resolution function, I can fully reproduce the top-1 accuracy.