Added missing argument in function call

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-04-17 13:49:20 +02:00
parent 2e388c0f82
commit 49115ccc1d

View File

@ -104,7 +104,8 @@ def _run_one_epoch_simple(dataset: tf.data.Dataset,
reconstruction_loss, x_decoded, z = _run_enc_dec_step_simple(encoder=encoder, reconstruction_loss, x_decoded, z = _run_enc_dec_step_simple(encoder=encoder,
decoder=decoder, decoder=decoder,
inputs=x, inputs=x,
global_step=global_step) global_step=global_step,
debug=debug)
enc_dec_loss_avg(reconstruction_loss) enc_dec_loss_avg(reconstruction_loss)
if int(global_step % train.LOG_FREQUENCY) == 0 and debug: if int(global_step % train.LOG_FREQUENCY) == 0 and debug: