2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Improve verbose printing for training

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-11 11:31:04 +02:00
parent e908248f89
commit f6f6a88c50
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/twomartens/masterthesis/ssd.py
View File

@ -498,9 +498,9 @@ def _train_epochs(nr_batches_per_epoch: int,
if verbose: if verbose:
print(( print((
f"[{epoch + 1:d}/{nr_epochs:d} - " f"[{epoch + 1:d}/{nr_epochs:d}] - "
f"train time: {outputs['per_epoch_time']:.2f}, " f"train time: {outputs['per_epoch_time']:.2f}, "
f"SSD loss: {outputs['ssd_loss']:.3f}, " f"SSD loss: {outputs['ssd_loss']:.3f}"
)) ))
# save weights at end of epoch # save weights at end of epoch