Update README.md
Browse files
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 |
```
|