Hi smth, thank you for your reply! The weird thing is that when running the same code on CPU (i.e., without .cuda()) everything is fine. I am trying to reproduce this with a small snippet. May I know if there’re any other possibilities that could cause this error?