Hi,
The cuda state on the GPU is quite big. Taking few hundreds of MB.
To avoid loading on the GPU a model saved on the gpu, you can do: torch.load('checkpoint_pretrain_secneflow.pth', map_location=torch.device('cpu')).