Update README.md

README.md

@@ -30,10 +30,12 @@ loss = MarginRankingLoss(margin=1)
 0% of margin 相当于准确率  
 ## 使用
 采用Pytorch加载，一般用加载一个CNN模型就可以使用，注意删除state_dict中的字典名字  
-  model_dict = torch.load('./checkpoint_5.pth')['state_dict']
+```
+  model_dict = torch.load('./checkpoint.pth')['state_dict']
   model_dict_mod = {}
   for key, value in model_dict.items():
     new_key = '.'.join(key.split('.')[1:])
     model_dict_mod[new_key] = value
   self.model.load_state_dict(model_dict_mod)
+```