From 54ed8d062633c355dd212e9e2b8833a8c2c9bf00 Mon Sep 17 00:00:00 2001 From: Double_V Date: Fri, 9 Jul 2021 11:38:01 +0800 Subject: [PATCH] fix ci --- tools/eval.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/tools/eval.py b/tools/eval.py index c99c7d47..8ae270b3 100755 --- a/tools/eval.py +++ b/tools/eval.py @@ -55,12 +55,8 @@ def main(): model = build_model(config['Architecture']) use_srn = config['Architecture']['algorithm'] == "SRN" - if "model_type" in config['Architecture'].keys(): - model_type = config['Architecture']['model_type'] - else: - model_type = None - best_model_dict = init_model(config, model, model_type) + best_model_dict = init_model(config, model) if len(best_model_dict): logger.info('metric in ckpt ***************') for k, v in best_model_dict.items(): -- GitLab