czzhangheng
/
DCRNN
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

converting to CPU

This commit is contained in:
Chintan Shah 2019-10-08 13:09:02 -04:00
parent f92e7295a0
commit d2913fd6f1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
model/pytorch/dcrnn_supervisor.py
View File

@ -131,8 +131,8 @@ class DCRNNSupervisor:
loss = self._compute_loss(y, output) loss = self._compute_loss(y, output)
losses.append(loss.item()) losses.append(loss.item())
y_truths.append(y) y_truths.append(y.cpu())
y_preds.append(output) y_preds.append(output.cpu())
mean_loss = np.mean(losses) mean_loss = np.mean(losses)