Update README.md
Browse files
README.md
CHANGED
|
@@ -35,7 +35,7 @@ loss = MarginRankingLoss(margin=1)
|
|
| 35 |
0% of margin 相当于准确率
|
| 36 |
## 使用
|
| 37 |
采用Pytorch加载，一般用加载一个CNN模型就可以使用，注意删除state_dict中的字典名字
|
| 38 |
-
```
|
| 39 |
model_dict = torch.load('./checkpoint.pth')['state_dict']
|
| 40 |
model_dict_mod = {}
|
| 41 |
for key, value in model_dict.items():
|
|
|
|
| 35 |
0% of margin 相当于准确率
|
| 36 |
## 使用
|
| 37 |
采用Pytorch加载，一般用加载一个CNN模型就可以使用，注意删除state_dict中的字典名字
|
| 38 |
+
```py
|
| 39 |
model_dict = torch.load('./checkpoint.pth')['state_dict']
|
| 40 |
model_dict_mod = {}
|
| 41 |
for key, value in model_dict.items():
|