VITA-Group / UVC

[ICLR 2022] "Unified Vision Transformer Compression" by Shixing Yu*, Tianlong Chen*, Jiayi Shen, Huan Yuan, Jianchao Tan, Sen Yang, Ji Liu, Zhangyang Wang
MIT License
45 stars 3 forks source link

Unable to reproduce the code. #3

Open wazenmai opened 1 year ago

wazenmai commented 1 year ago

Hello, I have read the paper and this is a great work on vision transformer compression!

However, when I want to reproduce the code following the command on github. I always get 0 accuracy after joint training. Could you provide the pre-trained model checkpoint so I can easily evaluate it on ImageNet?

I simply followed the command on github for setting the environment and run the stage 1 uvc training command. I installed apex library without global option. I used 2 NVIDIA GeForce RTX 2080 GPU to run it.

image

Best regards,