Hello author, I would like to ask whether this model can run on the 4090 machine? When I tried to reproduce this paper, there was a bug that I can't solve at present. I guess this may be due to the problem that my pytorch version is too new. I would like to ask if anyone can run this code on the 4090 machine? Here's a failure case I'm having, it happens during the evaluation phase

Hello author, I would like to ask whether this model can run on the 4090 machine? When I tried to reproduce this paper, there was a bug that I can't solve at present. I guess this may be due to the problem that my pytorch version is too new. I would like to ask if anyone can run this code on the 4090 machine? Here's a failure case I'm having, it happens during the evaluation phase