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

logging to 4 decimals

This commit is contained in:
Chintan Shah 2019-10-07 20:54:58 -04:00
parent f6e6713f74
commit 5d7694e293
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model/pytorch/dcrnn_supervisor.py
View File

@ -133,7 +133,7 @@ class DCRNNSupervisor:
per_timestep_loss /= num_batches per_timestep_loss /= num_batches
for i, val in enumerate(per_timestep_loss): for i, val in enumerate(per_timestep_loss):
self._logger.info("Dataset:{}, Timestep: {}, MAE:{}".format(dataset, i, val.item())) self._logger.info("Dataset:{}, Timestep: {}, MAE:{:.4f}".format(dataset, i, val.item()))
return mean_loss return mean_loss