Commit ·
f2bff8a
1
Parent(s): f291eba
Update README.md
Browse files
README.md
CHANGED
|
@@ -71,6 +71,7 @@ inputs = torch.randn(batch_size, seq_length)
|
|
| 71 |
|
| 72 |
# deterministic forecasting
|
| 73 |
with torch.no_grad():
|
|
|
|
| 74 |
outputs = model.generate(
|
| 75 |
inputs=inputs,
|
| 76 |
max_length=96,
|
|
@@ -82,6 +83,7 @@ with torch.no_grad():
|
|
| 82 |
|
| 83 |
# probabilistic forecasting
|
| 84 |
with torch.no_grad():
|
|
|
|
| 85 |
outputs = model.generate(
|
| 86 |
inputs=inputs,
|
| 87 |
max_length=96,
|
|
|
|
| 71 |
|
| 72 |
# deterministic forecasting
|
| 73 |
with torch.no_grad():
|
| 74 |
+
# output shape: [batch_size, 1, seq_len]
|
| 75 |
outputs = model.generate(
|
| 76 |
inputs=inputs,
|
| 77 |
max_length=96,
|
|
|
|
| 83 |
|
| 84 |
# probabilistic forecasting
|
| 85 |
with torch.no_grad():
|
| 86 |
+
# output shape: [batch_size, 100, seq_len]
|
| 87 |
outputs = model.generate(
|
| 88 |
inputs=inputs,
|
| 89 |
max_length=96,
|