提交 587e383e 编写于 作者: M malin10

bug fix

上级 854eddce
...@@ -520,7 +520,6 @@ class SingleInferRunner(RunnerBase): ...@@ -520,7 +520,6 @@ class SingleInferRunner(RunnerBase):
def run(self, context): def run(self, context):
self._dir_check(context) self._dir_check(context)
self.epoch_model_name_list.sort()
for index, epoch_name in enumerate(self.epoch_model_name_list): for index, epoch_name in enumerate(self.epoch_model_name_list):
for model_dict in context["phases"]: for model_dict in context["phases"]:
model_class = context["model"][model_dict["name"]]["model"] model_class = context["model"][model_dict["name"]]["model"]
......
...@@ -42,14 +42,14 @@ hyper_parameters: ...@@ -42,14 +42,14 @@ hyper_parameters:
gnn_propogation_steps: 1 gnn_propogation_steps: 1
# select runner by name # select runner by name
mode: [train_runner, infer_runner] mode: [single_cpu_train, single_cpu_infer]
# config of each runner. # config of each runner.
# runner is a kind of paddle training class, which wraps the train/infer process. # runner is a kind of paddle training class, which wraps the train/infer process.
runner: runner:
- name: train_runner - name: single_cpu_train
class: train class: train
# num of epochs # num of epochs
epochs: 2 epochs: 5
# device to run training or infer # device to run training or infer
device: cpu device: cpu
save_checkpoint_interval: 1 # save model interval of epochs save_checkpoint_interval: 1 # save model interval of epochs
...@@ -61,7 +61,7 @@ runner: ...@@ -61,7 +61,7 @@ runner:
init_model_path: "" # load model path init_model_path: "" # load model path
print_interval: 1 print_interval: 1
phases: [phase1] phases: [phase1]
- name: infer_runner - name: single_cpu_infer
class: infer class: infer
# device to run training or infer # device to run training or infer
device: cpu device: cpu
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册