MAX_PLOT_Y option

lvapeab · May 3, 2020 · 57978ab · 57978ab
1 parent 5879789
commit 57978ab
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/nmt_keras/build_callbacks.py b/nmt_keras/build_callbacks.py
@@ -88,6 +88,7 @@ def buildCallbacks(params, model, dataset):
                                               eval_on_epochs=params['EVAL_EACH_EPOCHS'],
                                               save_each_evaluation=params['SAVE_EACH_EVALUATION'],
                                               do_plot=params.get('PLOT_EVALUATION', False),
+                                              max_plot=params.get('MAX_PLOT_Y', None),
                                               verbose=params['VERBOSE'])
 
             callbacks.append(callback_metric)