thank you so much for this interested work and for sharing your code!
I run this code on:
pytorch=1.13.1
CUDA=11.7
but the line 101 in train.py:
model.cuda()
throws the following error: No CUDA GPUs are available, how could I solve this issue please :(
