From f4f2b58d25d289c9f93857f04a2a79e27642a086 Mon Sep 17 00:00:00 2001 From: z00478463 Date: Fri, 22 May 2020 15:49:38 +0800 Subject: [PATCH] for pylint 6th --- example/resnet50_imagenet2012_THOR/train.py | 17 ++++++----------- 1 file changed, 6 insertions(+), 11 deletions(-) diff --git a/example/resnet50_imagenet2012_THOR/train.py b/example/resnet50_imagenet2012_THOR/train.py index 421cda3886..020d56d55b 100644 --- a/example/resnet50_imagenet2012_THOR/train.py +++ b/example/resnet50_imagenet2012_THOR/train.py @@ -115,17 +115,12 @@ if __name__ == '__main__': step_size = dataset.get_dataset_size() loss_scale = FixedLossScaleManager(config.loss_scale, drop_overflow_update=False) - if config.lr_decay_mode == 'cosine': - lr = Tensor(warmup_cosine_annealing_lr(0.035, - step_size, - config.warmup_epochs, - 50, - config.T_max, - config.eta_min)) - else: - lr = Tensor(get_lr(global_step=0, lr_init=config.lr_init, lr_end=config.lr_end, lr_max=config.lr_max, - warmup_epochs=config.warmup_epochs, total_epochs=epoch_size, steps_per_epoch=step_size, - )) + lr = Tensor(warmup_cosine_annealing_lr(0.035, + step_size, + config.warmup_epochs, + 50, + config.T_max, + config.eta_min)) opt = THOR(filter(lambda x: x.requires_grad, net.get_parameters()), lr, config.momentum, damping, config.frequency, filter(lambda x: 'matrix_A' in x.name, net.get_parameters()),