Hello, When I train the model according to your instruction, the following error occurred when I executed the get_spix_gt function. How can I solve it?
PS: My Pytorch version is 1.9.0 and python version is 3.8 (But I don't think this bug is related to the version.)

