give me a moment, print(model)  does not work after   model.load_state_dict(checkpoint) because the error is due to checkpoint variable itself.

Traceback (most recent call last):
File “predict.py”, line 29, in 
model.load_state_dict(checkpoint[‘model_state_dict’])
KeyError: ‘model_state_dict’