(config, parameters,lr)
| 77 | |
| 78 | |
| 79 | def get_optimizer(config, parameters,lr): |
| 80 | if config.optim.optimizer == 'Adam': |
| 81 | return optim.Adam(parameters, lr=lr, weight_decay=config.optim.weight_decay, |
| 82 | betas=(config.optim.beta1, 0.999), amsgrad=config.optim.amsgrad, |
| 83 | eps=config.optim.eps) |
| 84 | elif config.optim.optimizer == 'RMSProp': |
| 85 | return optim.RMSprop(parameters, lr=lr, weight_decay=config.optim.weight_decay) |
| 86 | elif config.optim.optimizer == 'SGD': |
| 87 | return optim.SGD(parameters, lr=lr, momentum=0.9) |
| 88 | else: |
| 89 | raise NotImplementedError( |
| 90 | 'Optimizer {} not understood.'.format(config.optim.optimizer)) |
| 91 | |
| 92 | def get_model(config, is_train = True, resume = False, resume_path = None): |
| 93 | neighbour_matrix = get_neighbour_matrix_from_hand(parents,childrens,num_joints=config.hyponet.num_joints,num_edges=config.hyponet.num_twists,knn=config.hyponet.knn) |
no outgoing calls
no test coverage detected