Update README.md
Browse files
README.md
CHANGED
|
@@ -74,7 +74,6 @@ u = ds["u100"].values[0]
|
|
| 74 |
v = ds["v100"].values[0]
|
| 75 |
inputs = torch.from_numpy(np.stack([u, v], axis=0)).unsqueeze(0).float() # shape (1, 2, H, W)
|
| 76 |
|
| 77 |
-
|
| 78 |
# prediction
|
| 79 |
outputs = model(inputs)
|
| 80 |
|
|
|
|
| 74 |
v = ds["v100"].values[0]
|
| 75 |
inputs = torch.from_numpy(np.stack([u, v], axis=0)).unsqueeze(0).float() # shape (1, 2, H, W)
|
| 76 |
|
|
|
|
| 77 |
# prediction
|
| 78 |
outputs = model(inputs)
|
| 79 |
|