diff --git a/example/resnet50_imagenet2012_THOR/train.py b/example/resnet50_imagenet2012_THOR/train.py
index 020d56d55b..3843338a51 100644
--- a/example/resnet50_imagenet2012_THOR/train.py
+++ b/example/resnet50_imagenet2012_THOR/train.py
@@ -33,7 +33,7 @@ import numpy as np
 from config_imagenet import config
 from crossentropy import CrossEntropy
 from dataset_imagenet import create_dataset
-from lr_generator import get_lr, warmup_cosine_annealing_lr
+from lr_generator import  warmup_cosine_annealing_lr
 
 random.seed(1)
 np.random.seed(1)