From fcc1b857d402d9cf1f8d12edc7c00472f2f14b07 Mon Sep 17 00:00:00 2001 From: cuicheng01 Date: Thu, 10 Jun 2021 08:40:05 +0000 Subject: [PATCH] fix eval_interval bugs --- ppcls/data/__init__.py | 6 +++--- ppcls/engine/trainer.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/ppcls/data/__init__.py b/ppcls/data/__init__.py index eaa3aee67..0e8160b05 100644 --- a/ppcls/data/__init__.py +++ b/ppcls/data/__init__.py @@ -35,13 +35,13 @@ from ppcls.data.preprocess import transform def build_dataloader(config, mode, device, seed=None): - assert mode in ['Train', 'Eval', 'Test', 'Gallery', 'Query' - ], "Mode should be Train, Eval, Test, Gallery or Query" + assert mode in ['Train', 'Eval', 'Test', + ], "Mode should be Train, Eval, Test" # build dataset config_dataset = config[mode]['dataset'] config_dataset = copy.deepcopy(config_dataset) dataset_name = config_dataset.pop('name') - if 'batch_transform_ops' in config_dataset: + if 'batch_transform_ops' in config_dataset:h batch_transform = config_dataset.pop('batch_transform_ops') else: batch_transform = None diff --git a/ppcls/engine/trainer.py b/ppcls/engine/trainer.py index 837493578..43a5f63da 100644 --- a/ppcls/engine/trainer.py +++ b/ppcls/engine/trainer.py @@ -223,7 +223,7 @@ class Trainer(object): # eval model and save model if possible if self.config["Global"][ "eval_during_train"] and epoch_id % self.config["Global"][ - "eval_during_train"] == 0: + "eval_interval"] == 0: acc = self.eval(epoch_id) if acc > best_metric["metric"]: best_metric["metric"] = acc