def get_solver(param): solver = SolverParameter() solver.net = param['file_train_val_net'] solver.test_interval = param['solver_test_interval'] solver.base_lr = param['solver_base_lr'] solver.weight_decay = param['solver_weight_decay'] solver.lr_policy = param['solver_lr_policy'] solver.display = param['solver_display'] solver.max_iter = param['solver_max_iter'] #solver.clip_gradients = param['solver_clip_gradients'] solver.snapshot = param['solver_snapshot'] solver.lr_policy = param['solver_lr_policy'] solver.stepsize = param['solver_stepsize'] solver.gamma = param['solver_gamma'] solver.snapshot_prefix = param['solver_snapshot_prefix'] solver.random_seed = param['solver_random_seed'] solver.solver_mode = param['solver_solver_mode'] solver.device_id = param['device_id'] solver.test_iter.append(param['solver_test_iter']) return solver