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

Added additional values to print

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

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

@ -497,7 +497,9 @@ def _train_epochs(nr_batches_per_epoch: int,
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}, "
f"batch_counter: {batch_counter:d}, "
f"nr_batches_per_epoch: {nr_batches_per_epoch:d}"
)) ))
# save weights at end of epoch # save weights at end of epoch