Line of error: model.load_state_dict(checkpoint['net']) Error: model.load_state_dict(checkpoint['net']) KeyError: 'net'