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

bug fix

上级 854eddce
......@@ -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"]
......
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册