diff --git a/nemo/core/neural_factory.py b/nemo/core/neural_factory.py index 8aecda3d64d9..4402ded7b927 100644 --- a/nemo/core/neural_factory.py +++ b/nemo/core/neural_factory.py @@ -62,7 +62,7 @@ class OperationMode(Enum): """Training or Inference (Evaluation) mode""" training = 0 - inference = 1 + evaluation = 1 both = 2