diff --git a/example/resnet50_imagenet2012/train.py b/example/resnet50_imagenet2012/train.py
index 1992bfda95f875cb379e384c7c39a71011b60772..d87189f8a7f4994cdb5e8f54a3f44a69b35693c4 100755
--- a/example/resnet50_imagenet2012/train.py
+++ b/example/resnet50_imagenet2012/train.py
@@ -86,7 +86,9 @@ if __name__ == '__main__':
         opt = Momentum(filter(lambda x: x.requires_grad, net.get_parameters()), lr, config.momentum,
                        config.weight_decay, config.loss_scale)
 
-        model = Model(net, loss_fn=loss, optimizer=opt, loss_scale_manager=loss_scale, metrics={'acc'})
+        model = Model(net, loss_fn=loss, optimizer=opt, loss_scale_manager=loss_scale, metrics={'acc'}, amp_level="O2",
+                      keep_batchnorm_fp32=False)
+
 
         time_cb = TimeMonitor(data_size=step_size)
         loss_cb = LossMonitor()