fixed range bug

2019-10-07 20:40:54 -04:00 · 2019-10-07 20:40:54 -04:00 · f6e6713f74
parent 2560e1d954
commit f6e6713f74
1 changed files with 1 additions and 1 deletions
--- a/model/pytorch/dcrnn_supervisor.py
+++ b/model/pytorch/dcrnn_supervisor.py
@ -118,7 +118,7 @@ class DCRNNSupervisor:
                output = self.dcrnn_model(x)

                # (horizon, batch_size, num_sensor * output_dim)
-                for t in y.size(0):
+                for t in range(y.size(0)):
                    per_timestep_loss[t] += self._compute_loss(y[t], output[t])

                num_batches += 1