Update README.md
Browse files
README.md
CHANGED
|
@@ -32,7 +32,6 @@ time_series = torch.randn(batch_size, lookback_length, channels)
|
|
| 32 |
device = torch.cuda.current_device() if torch.cuda.is_available() else 'cpu'
|
| 33 |
model = model.to(device)
|
| 34 |
time_series = time_series.to(device)
|
| 35 |
-
print(f'device: {device}')
|
| 36 |
|
| 37 |
# Generate forecasts
|
| 38 |
forecast_length = 96
|
|
|
|
| 32 |
device = torch.cuda.current_device() if torch.cuda.is_available() else 'cpu'
|
| 33 |
model = model.to(device)
|
| 34 |
time_series = time_series.to(device)
|
|
|
|
| 35 |
|
| 36 |
# Generate forecasts
|
| 37 |
forecast_length = 96
|