Sorry for the late update. These days we utilize these code as our base for other downstream tasks. We don't have time to release them.
For the hardware implementation issues, please contact us. We are pretty happy to talk about the hardware efficiency on resource-limited devices.
Python 3.8 Pytorch 1.7.1 Torchvision 0.8.2 timm 0.4.12
https://drive.google.com/drive/folders/19c7s_AglQGzVWfvzySpCdBKWqqPabxmN?usp=drive_link