From b30d74540190b2e4e05798585f99eee2514d7c8d Mon Sep 17 00:00:00 2001 From: Richard Liaw Date: Fri, 7 Aug 2020 20:29:04 -0700 Subject: [PATCH] [tune] reorder transformer example save (#9994) Co-authored-by: Amog Kamsetty --- python/ray/tune/examples/pbt_transformers/trainer.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/python/ray/tune/examples/pbt_transformers/trainer.py b/python/ray/tune/examples/pbt_transformers/trainer.py index 25d87535a..79c193a08 100644 --- a/python/ray/tune/examples/pbt_transformers/trainer.py +++ b/python/ray/tune/examples/pbt_transformers/trainer.py @@ -18,6 +18,7 @@ logger = logging.getLogger(__name__) The only changes to the original transformers.Trainer are: - Report eval metrics to Tune - Save state using Tune's checkpoint directories + - Pass in extra arguments for wandb """ @@ -39,10 +40,8 @@ class TuneTransformerTrainer(transformers.Trainer): output = self._prediction_loop( eval_dataloader, description="Evaluation") self._log(output.metrics) - - tune.report(**output.metrics) - self.save_state() + tune.report(**output.metrics) return output.metrics