diff --git a/pytorch_lightning/core/step_result.py b/pytorch_lightning/core/step_result.py index 612f8ee3b2..62533a878c 100644 --- a/pytorch_lightning/core/step_result.py +++ b/pytorch_lightning/core/step_result.py @@ -497,8 +497,9 @@ class TrainResult(Result): return result Args: - early_stop_on: - checkpoint_on: + minimize: Metric currently being minimized. + early_stop_on: Metric to early stop on. + checkpoint_on: Metric to checkpoint on. hiddens: """ @@ -652,8 +653,8 @@ class EvalResult(Result): return result Args: - early_stop_on: - checkpoint_on: + early_stop_on: Metric to early stop on. + checkpoint_on: Metric to checkpoint on. hiddens: """