From 7ecfd76cc05b23522f8bb6f049bd12ec02ea4710 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Thu, 26 May 2022 07:11:31 -0700 Subject: [PATCH] fix evaluation config splat in training decoder script --- train_decoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_decoder.py b/train_decoder.py index af72bea..a4651d8 100644 --- a/train_decoder.py +++ b/train_decoder.py @@ -347,7 +347,7 @@ def train( # Compute evaluation metrics if exists(evaluate_config): print(print_ribbon(f"Starting Evaluation {epoch}", repeat=40)) - evaluation = evaluate_trainer(trainer, dataloaders["val"], inference_device, **evaluate_config) + evaluation = evaluate_trainer(trainer, dataloaders["val"], inference_device, **evaluate_config.dict()) tracker.log(evaluation, step=step, verbose=True) # Generate sample images