Loading a checkpoint needs weights_only=False

[ghsanti]

📚 Documentation

Here in loading a checkpoint

model = TheModelClass(*args, **kwargs)
optimizer = TheOptimizerClass(*args, **kwargs)

checkpoint = torch.load(PATH, weights_only=True)
model.load_state_dict(checkpoint['model_state_dict'])
optimizer.load_state_dict(checkpoint['optimizer_state
#...

One should load all the model not weights only:

checkpoint = torch.load(PATH, weights_only=True)

to

checkpoint = torch.load(PATH, weights_only=False)

[Tony-Y]

You should carefully read https://pytorch.org/docs/stable/generated/torch.load.html. It is insecure to set weights_only to False.