From 2682384775ffb5b2c1472f060f1eb801f7ae80b4 Mon Sep 17 00:00:00 2001 From: littletomatodonkey Date: Thu, 16 Apr 2020 08:11:47 +0000 Subject: [PATCH] fix arch --- ppcls/utils/save_load.py | 2 +- tools/train.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ppcls/utils/save_load.py b/ppcls/utils/save_load.py index 302171767..fef1b8164 100644 --- a/ppcls/utils/save_load.py +++ b/ppcls/utils/save_load.py @@ -106,7 +106,7 @@ def load_params(exe, prog, path, ignore_params=[]): fluid.io.set_program_state(prog, state) -def init_model(config, program, exe, prefix="ppcls"): +def init_model(config, program, exe, prefix=""): """ load model from checkpoint or pretrained_model """ diff --git a/tools/train.py b/tools/train.py index 3a5afd6ce..6f4a099b2 100644 --- a/tools/train.py +++ b/tools/train.py @@ -96,7 +96,7 @@ def main(args): if epoch_id % config.save_interval == 0: model_path = os.path.join(config.model_save_dir, - config.architecture) + config.ARCHITECTURE) save_model(train_prog, model_path, epoch_id)