We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a1445e4 commit 5b89324Copy full SHA for 5b89324
train.py
@@ -150,8 +150,9 @@ def train(opt):
150
# Write validation result into summary
151
if tf is not None:
152
add_summary_value(tf_summary_writer, 'validation loss', val_loss, iteration)
153
- for k,v in lang_stats.items():
154
- add_summary_value(tf_summary_writer, k, v, iteration)
+ if opt.language_eval == 1:
+ for k,v in lang_stats.items():
155
+ add_summary_value(tf_summary_writer, k, v, iteration)
156
tf_summary_writer.flush()
157
val_result_history[iteration] = {'loss': val_loss, 'lang_stats': lang_stats, 'predictions': predictions}
158
0 commit comments