Example #1
0
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