converting to CPU

2019-10-08 13:09:02 -04:00 · 2019-10-08 13:09:02 -04:00 · d2913fd6f1
parent f92e7295a0
commit d2913fd6f1
1 changed files with 2 additions and 2 deletions
--- a/model/pytorch/dcrnn_supervisor.py
+++ b/model/pytorch/dcrnn_supervisor.py
@ -131,8 +131,8 @@ class DCRNNSupervisor:
                loss = self._compute_loss(y, output)
                losses.append(loss.item())

-                y_truths.append(y)
-                y_preds.append(output)
+                y_truths.append(y.cpu())
+                y_preds.append(output.cpu())

            mean_loss = np.mean(losses)