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.test_iter.append(param['solver_test_iter'])
return solver
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.test_iter.append(param["solver_test_iter"])
return solver
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.test_iter.append(param['solver_test_iter'])
return solver
