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

Added resetting the states of the metrics

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-02-08 06:59:57 +01:00
parent a6e90e70b3
commit 4dda663e3b
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -198,6 +198,12 @@ def train_mnist(folding_id: int, inlier_classes: Sequence[int], total_classes: i
summary_ops_v2.scalar('encoder_decoder_loss', enc_dec_loss_avg.result(), step=global_step_enc_dec) summary_ops_v2.scalar('encoder_decoder_loss', enc_dec_loss_avg.result(), step=global_step_enc_dec)
summary_ops_v2.scalar('z_discriminator_loss', zd_loss_avg.result(), step=global_step_zd) summary_ops_v2.scalar('z_discriminator_loss', zd_loss_avg.result(), step=global_step_zd)
summary_ops_v2.scalar('x_discriminator_loss', xd_loss_avg.result(), step=global_step_xd) summary_ops_v2.scalar('x_discriminator_loss', xd_loss_avg.result(), step=global_step_xd)
# reset the metrics states
encoder_loss_avg.reset_states()
decoder_loss_avg.reset_states()
enc_dec_loss_avg.reset_states()
zd_loss_avg.reset_states()
xd_loss_avg.reset_states()
if it == 0: if it == 0:
directory = 'results' + str(inlier_classes[0]) directory = 'results' + str(inlier_classes[0])