Python set_opt_param示例

示例#1

 def _init_optimizer(self, optimizers):
     self.optimizers = optimizers
     # reinitilize schedulers
     self.schedulers = []
     for optimizer in self.optimizers:
         util.set_opt_param(optimizer, 'initial_lr', self.opt.lr)
         util.set_opt_param(optimizer, 'weight_decay', self.opt.wd)

示例#2

文件： train_errnet_unaligned.py 项目： mattcoldwater/ERRNet

def set_learning_rate(lr):
    for optimizer in engine.model.optimizers:
        util.set_opt_param(optimizer, 'lr', lr)

示例#3

文件： errnet_model.py 项目： siju-samuel/reflection-removal

 def _init_optimizer(self, optimizers):
     self.optimizers = optimizers
     for optimizer in self.optimizers:
         util.set_opt_param(optimizer, 'initial_lr', self.opt.lr)
         util.set_opt_param(optimizer, 'weight_decay', self.opt.wd)

示例#4

def set_learning_rate(lr):
    for optimizer in engine.model.optimizers:
        print('[i] set learning rate to {}'.format(lr))
        util.set_opt_param(optimizer, 'lr', lr)