Reproduce ConTNet-Tiny*

[akuxcw]

Hi, thank you for sharing this awesome work! I'm trying to reproduce the result of ConTNet-Tiny*(74.9) with this code. But my result is very low (60.85). I used the following command:

CUDA_VISIBLE_DEVICES=0,1,2,3 python3 -m torch.distributed.launch --nproc_per_node=4 --master_port 29501 main.py --arch ConT-Ti --batch_size 256 --save_path contnet-tiny --save_best True 

Is there any suggestion to correctly reproduce the results?

[yan-hao-tian]

That's weird. In general, the top-1 acc. of CONTNET-TI on ImageNet val can reach about %60 at very early epoch. Thanks for your attention and I will check it.