Update pytorch_training.py

AIStream-Peelout · Aug 5, 2020 · 0cae9d5 · 0cae9d5
1 parent 204e7c3
commit 0cae9d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flood_forecast/pytorch_training.py b/flood_forecast/pytorch_training.py
@@ -186,7 +186,7 @@ def compute_validation(validation_loader: DataLoader,
                         :,
                         0]
                 else:
-                    output = simple_decode(model, src, targ.shape[1], targ, 1)[:, :, 0]
+                    output = simple_decode(model, src, targ.shape[1], targ, sequence_size)[:, :, 0]
             else:
                 output = model(src.float())
             labels = targ[:, :, 0]