mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #820 from cuicheng01/develop_reg

Browse Source 
fix eval_interval bugs
This commit is contained in:
Wei Shengyu 2021-06-10 16:56:50 +08:00 committed by GitHub
commit 4905424f31
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ppcls/data/__init__.py
View File

@ -35,8 +35,8 @@ from ppcls.data.preprocess import transform
def build_dataloader(config, mode, device, seed=None): def build_dataloader(config, mode, device, seed=None):
assert mode in ['Train', 'Eval', 'Test', 'Gallery', 'Query' assert mode in ['Train', 'Eval', 'Test',
], "Mode should be Train, Eval, Test, Gallery or Query" ], "Mode should be Train, Eval, Test"
# build dataset # build dataset
config_dataset = config[mode]['dataset'] config_dataset = config[mode]['dataset']
config_dataset = copy.deepcopy(config_dataset) config_dataset = copy.deepcopy(config_dataset)

2
ppcls/engine/trainer.py
View File

@ -223,7 +223,7 @@ class Trainer(object):
# eval model and save model if possible # eval model and save model if possible
if self.config["Global"][ if self.config["Global"][
"eval_during_train"] and epoch_id % self.config["Global"][ "eval_during_train"] and epoch_id % self.config["Global"][
"eval_during_train"] == 0: "eval_interval"] == 0:
acc = self.eval(epoch_id) acc = self.eval(epoch_id)
if acc > best_metric["metric"]: if acc > best_metric["metric"]:
best_metric["metric"] = acc best_metric["metric"] = acc