Wait something’s fishy - it was working when pushing variables to cuda and running pytorch, but now gives the same error. In the meantime the computer went to sleep. Maybe I need to restart my system. Give me a minute …