I save and load all my models using the following functions inside a model class:
def save_params(self, exDir):
	print 'saving params...'
	torch.save(self.state_dict(), join(exDir, 'dae_params'))

def load_params(self, exDir):
	print 'loading params...'
	self.load_state_dict(torch.load(join(exDir, 'dae_params')))

Normally the models save and load without error. However, I am currently getting the following error when trying to load model:
THCudaCheck FAIL file=/pytorch/torch/lib/THC/generic/THCStorage.c line=82 error=46 : all CUDA-capable devices are busy or unavailable
Segmentation fault
The model is of size 35M and GPU is of size 11439M. Any suggestions as to why this may be happening? I have checked (w/ nvidia-smi and there are GPUs available, and I am still able to load other models of the same size on the same GPU).
Thanks in advance,
Toni.