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

logging and refactor

This commit is contained in:
Chintan Shah 2019-10-06 14:31:46 -04:00
parent 31acadedce
commit 036e552bf6
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

@ -154,7 +154,7 @@ class DCRNNSupervisor:
torch.nn.utils.clip_grad_norm_(self.dcrnn_model.parameters(), self.max_grad_norm) torch.nn.utils.clip_grad_norm_(self.dcrnn_model.parameters(), self.max_grad_norm)
optimizer.step() optimizer.step()
self._logger.info("finished one batch in {}".format(time.time() - start_time)) self._logger.info("finished one batch in {:.1f}s".format(time.time() - start_time))
lr_scheduler.step() lr_scheduler.step()