Update README.md
Browse files
README.md
CHANGED
|
@@ -64,6 +64,7 @@ prediction = torch.nn.functional.interpolate(
|
|
| 64 |
mode="bicubic",
|
| 65 |
align_corners=False,
|
| 66 |
)
|
|
|
|
| 67 |
|
| 68 |
# visualize the prediction
|
| 69 |
output = prediction.squeeze().cpu().numpy()
|
|
|
|
| 64 |
mode="bicubic",
|
| 65 |
align_corners=False,
|
| 66 |
)
|
| 67 |
+
```
|
| 68 |
|
| 69 |
# visualize the prediction
|
| 70 |
output = prediction.squeeze().cpu().numpy()
|