diff --git a/core/model.py b/core/model.py index bb2040e44458d5a402d54bfc780ae501c9c9d06d..e96d01fe0bbcc0541a2b5aea458e2cb34fa89303 100755 --- a/core/model.py +++ b/core/model.py @@ -138,9 +138,7 @@ class ModelBase(object): os.environ["FLAGS_communicator_is_sgd_optimizer"] = '0' if name == "SGD": - reg = envs.get_global_env("hyper_parameters.reg", 0.0001) - optimizer_i = fluid.optimizer.SGD( - lr, regularization=fluid.regularizer.L2DecayRegularizer(reg)) + optimizer_i = fluid.optimizer.SGD(lr) elif name == "ADAM": optimizer_i = fluid.optimizer.Adam(lr, lazy_mode=True) elif name == "ADAGRAD": 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