diff --git a/core/trainers/framework/runner.py b/core/trainers/framework/runner.py index 5da1f5df723d43dcefd327193e0b7e7ab5368366..277b2ac7bbd1ad42f40f3501f6497fc471b762f6 100644 --- a/core/trainers/framework/runner.py +++ b/core/trainers/framework/runner.py @@ -520,7 +520,6 @@ class SingleInferRunner(RunnerBase): def run(self, context): self._dir_check(context) - self.epoch_model_name_list.sort() for index, epoch_name in enumerate(self.epoch_model_name_list): for model_dict in context["phases"]: model_class = context["model"][model_dict["name"]]["model"] diff --git a/models/recall/gnn/config.yaml b/models/recall/gnn/config.yaml index dd9c1728ab9a810b82ec0076913557f1486ac2ce..5573bdc3dbefcb16d7ab86212bc4dfcac4a72a87 100755 --- a/models/recall/gnn/config.yaml +++ b/models/recall/gnn/config.yaml @@ -42,14 +42,14 @@ hyper_parameters: gnn_propogation_steps: 1 # select runner by name -mode: [train_runner, infer_runner] +mode: [single_cpu_train, single_cpu_infer] # config of each runner. # runner is a kind of paddle training class, which wraps the train/infer process. runner: -- name: train_runner +- name: single_cpu_train class: train # num of epochs - epochs: 2 + epochs: 5 # device to run training or infer device: cpu save_checkpoint_interval: 1 # save model interval of epochs @@ -61,7 +61,7 @@ runner: init_model_path: "" # load model path print_interval: 1 phases: [phase1] -- name: infer_runner +- name: single_cpu_infer class: infer # device to run training or infer device: cpu