fix arch

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
     """

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)