diff --git a/official/cv/resnet/train.py b/official/cv/resnet/train.py
index 166146ee7f956abb5be69a44e01dd1f1ab563598..b09f6a18e70dcb37025870afd1869f1df2f30765 100644
--- a/official/cv/resnet/train.py
+++ b/official/cv/resnet/train.py
@@ -133,8 +133,6 @@ def set_parameter():
     if target == "CPU":
         config.run_distribute = False
 
-    config.save_graphs = not config.pre_trained
-
     # init context
     if config.mode_name == 'GRAPH':
         if target == "Ascend":