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

Added formatting of decoder loss

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-02-08 13:41:09 +01:00
parent a0094266b3
commit 948ef64423
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/aae/train.py
View File

@ -227,7 +227,7 @@ def train_mnist(folding_id: int, inlier_classes: Sequence[int], total_classes: i
print(( print((
f"[{epoch + 1:d}/{train_epoch:d}] - " f"[{epoch + 1:d}/{train_epoch:d}] - "
f"train time: {per_epoch_time:.2f}, " f"train time: {per_epoch_time:.2f}, "
f"Decoder loss: {decoder_loss}, " f"Decoder loss: {decoder_loss:.3f}, "
f"X Discriminator loss: {xd_loss:.3f}, " f"X Discriminator loss: {xd_loss:.3f}, "
f"Z Discriminator loss: {zd_loss:.3f}, " f"Z Discriminator loss: {zd_loss:.3f}, "
f"Encoder + Decoder loss: {enc_dec_loss:.3f}, " f"Encoder + Decoder loss: {enc_dec_loss:.3f}, "