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():