Commit b5fd08f1 authored by IamTao's avatar IamTao

update the code in the public repo (remove useless optimizer).

parent 1b63b1cc
......@@ -5,17 +5,9 @@ from pcode.optim.dgc import DGC
from pcode.optim.parallel_choco import ParallelCHOCO
from pcode.optim.parallel_choco_v import ParallelCHOCO_V
from pcode.optim.ef_sign_sgd import EF_SignSGD
from pcode.optim.others.sign_sgd import SignSGD
from pcode.optim.others.local_sign_sgd import Local_SignSGD
from pcode.optim.others.local_sgd import LocalSGD
from pcode.optim.dcd_psgd import DCD_PSGD
from pcode.optim.ecd_psgd import ECD_PSGD
from pcode.optim.deep_squeeze import DeepSqueeze
from pcode.optim.others.local_ef_sign_sgd_v import Local_EFSignSGD_V
from pcode.optim.others.local_sign_sgd_v import Local_SignSGD_V
from pcode.optim.others.sign_sgd_v import SignSGD_V
from pcode.optim.others.adam import Adam
from pcode.optim.others.local_adam import LocalAdam
def define_optimizer(conf, model):
......@@ -46,24 +38,8 @@ def define_optimizer(conf, model):
optim_class = ParallelCHOCO_V
elif conf.optimizer == "ef_sign_sgd":
optim_class = EF_SignSGD
elif conf.optimizer == "sign_sgd":
optim_class = SignSGD
elif conf.optimizer == "local_sign_sgd":
optim_class = Local_SignSGD
elif conf.optimizer == "local_sgd":
optim_class = LocalSGD
elif conf.optimizer == "deep_squeeze":
optim_class = DeepSqueeze
elif conf.optimizer == "sign_sgd_v":
optim_class = SignSGD_V
elif conf.optimizer == "local_sign_sgd_v":
optim_class = Local_SignSGD_V
elif conf.optimizer == "local_ef_sign_sgd_v":
optim_class = Local_EFSignSGD_V
elif conf.optimizer == "adam":
optim_class = Adam
elif conf.optimizer == "local_adam":
optim_class = LocalAdam
else:
raise NotImplementedError
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment