@JuanFMontesinos,
I added two lines around the ‘load_state_dict’ line:
        predictor = predictor.cpu()
        predictor.load_state_dict(torch.load(dirpath+'/StatePredictor.pth'))         # Load state_dict
        predictor = predictor.cuda()

It barely helps.  Instead of being able to use a max vector length of 4900, now it’s 5000.  The problems are pushed out to 5100.