diff --git a/models/rank/dnn/config.yaml b/models/rank/dnn/config.yaml index 38d841655dffa9a392ee6c7cd416b99de4e15b5d..539cbb00fbb83197b120238a584ae13d348cc49e 100755 --- a/models/rank/dnn/config.yaml +++ b/models/rank/dnn/config.yaml @@ -71,22 +71,6 @@ runner: print_interval: 10 phases: [phase1] -- name: single_gpu_train - class: train - # num of epochs - epochs: 4 - # device to run training or infer - device: gpu - selected_gpus: "2" - save_checkpoint_interval: 2 # save model interval of epochs - save_inference_interval: 4 # save inference - save_checkpoint_path: "increment" # save checkpoint path - save_inference_path: "inference" # save inference path - save_inference_feed_varnames: [] # feed vars of save inference - save_inference_fetch_varnames: [] # fetch vars of save inference - init_model_path: "" # load model path - print_interval: 10 - - name: single_cpu_infer class: infer # num of epochs @@ -96,33 +80,6 @@ runner: init_model_path: "increment/0" # load model path phases: [phase2] -- name: local_cluster_cpu_ps_train - class: local_cluster - epochs: 4 - device: cpu - save_checkpoint_interval: 2 # save model interval of epochs - save_inference_interval: 4 # save inference - save_checkpoint_path: "increment" # save checkpoint path - save_inference_path: "inference" # save inference path - save_inference_feed_varnames: [] # feed vars of save inference - save_inference_fetch_varnames: [] # fetch vars of save inference - init_model_path: "" # load model path - print_interval: 1 - -- name: multi_gpu_train - class: train - epochs: 4 - device: gpu - selected_gpus: "2,3" - save_checkpoint_interval: 2 # save model interval of epochs - save_inference_interval: 4 # save inference - save_checkpoint_path: "increment" # save checkpoint path - save_inference_path: "inference" # save inference path - save_inference_feed_varnames: [] # feed vars of save inference - save_inference_fetch_varnames: [] # fetch vars of save inference - init_model_path: "" # load model path - print_interval: 10 - # runner will run all the phase in each epoch phase: - name: phase1