@JuanFMontesinos, Thank You!
I put it like this:
        state_dict = torch.load(dirpath+'/StatePredictor.pth', map_location=lambda storage, loc: storage)
        predictor.load_state_dict(state_dict)

Does that look right?
It certainly doesn’t have a memory leak now.