godx7 commited on
Commit
973778b
·
verified ·
1 Parent(s): c80d924

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -4,7 +4,7 @@ The pth file contains model architecture and weights, so you should change the w
4
 
5
  ```python
6
  # model = R2RC(config) do not need this
7
- model = torch.load('xxx.pth')
8
  model.to(device)
9
  model.eval()
10
  ```
 
4
 
5
  ```python
6
  # model = R2RC(config) do not need this
7
+ model = torch.load('xxx.pth') # make sure you've downloaded the xxx.pth to your local directory
8
  model.to(device)
9
  model.eval()
10
  ```