reproduction on GPU device

[bambooCV]

Thanks for great job. I would like to reproduce crossformer on GPU devices.
I followed all instrution that your rep provided. However training was sooooo slow. Is that any problem in my env? Could you help me to solve this problem?

[mhyatt000]

@bambooCV see my issue: #2
I think this will help you solve your problem