提交 95265835 编写于 作者: F frankwhzhang

fix esmm

上级 e0d38aeb
......@@ -36,7 +36,7 @@ hyper_parameters:
strategy: async
#use infer_runner mode and modify 'phase' below if infer
mode: train_runner
mode: [train_runner]
#mode: infer_runner
runner:
......@@ -49,17 +49,19 @@ runner:
save_checkpoint_path: "increment"
save_inference_path: "inference"
print_interval: 10
phases: [train]
- name: infer_runner
class: single_infer
init_model_path: "increment/0"
device: cpu
epochs: 3
epochs: 1
phases: [infer]
phase:
# - name: train
# model: "{workspace}/model.py"
# dataset_name: dataset_train
# thread_num: 1
- name: train
model: "{workspace}/model.py"
dataset_name: dataset_train
thread_num: 1
- name: infer
model: "{workspace}/model.py"
dataset_name: dataset_infer
......
此差异已折叠。
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册