7109029214

Replace ```python res18.load_state_dict(checkpoint['model_state_dict']) ``` To ```python res18.load_state_dict(checkpoint['state_dict']) ```