TypeError: cannot pickle 'WeakMethod' object

[Naclzno]

Environment: pytorch2.0.1
I would like to ask if this problem is caused by the pytorch version
source code(trainer.py):
def save_checkpoint(to_save, logdir, suffix=""):
torch.save(to_save, logdir +"/FourDenoiseNet" + suffix)

save_checkpoint(state, self.log, suffix="")

[Naclzno]

I solved the problem by switching pytorch version to 1.2.