From f6e6713f74bea45f345b2a78590e50a5c5e784c2 Mon Sep 17 00:00:00 2001 From: Chintan Shah Date: Mon, 7 Oct 2019 20:40:54 -0400 Subject: [PATCH] fixed range bug --- model/pytorch/dcrnn_supervisor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model/pytorch/dcrnn_supervisor.py b/model/pytorch/dcrnn_supervisor.py index af5732d..c09052e 100644 --- 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