DELETE STATE DICT work for me
state_dict = torch.load(state_dict_path) 
model.load_state_dict(state_dict)
del state_dict

But I don’t know if the weights in the model would be deleted after del state_dict.