From d2913fd6f19d36a770ab33a5d0d22fe064a1b8da Mon Sep 17 00:00:00 2001 From: Chintan Shah Date: Tue, 8 Oct 2019 13:09:02 -0400 Subject: [PATCH] converting to CPU --- model/pytorch/dcrnn_supervisor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model/pytorch/dcrnn_supervisor.py b/model/pytorch/dcrnn_supervisor.py index 0497323..c2a34d4 100644 --- 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)