Change config

Signed-off-by: MaximumEntropy <[email protected]>
NVIDIA · Mar 19, 2022 · 8968a5b · 8968a5b
1 parent 9d9bd77
commit 8968a5b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/nlp/machine_translation/conf/aayn_base_megatron.yaml b/examples/nlp/machine_translation/conf/aayn_base_megatron.yaml
@@ -31,6 +31,7 @@ exp_manager:
     monitor: val_loss
     save_top_k: 10
     mode: min
+    save_best_model: True
     always_save_nemo: False # saves nemo file during validation, not implemented for model parallel
     filename: 'megatron_nmt--{val_loss:.3f}-{step}'
     model_parallel_size: ${multiply:${model.tensor_model_parallel_size}, ${model.pipeline_model_parallel_size}}