diff --git a/mindspore/train/callback/_dataset_graph.py b/mindspore/train/callback/_dataset_graph.py
index 224d53b27ea..4a9914a2169 100644
--- a/mindspore/train/callback/_dataset_graph.py
+++ b/mindspore/train/callback/_dataset_graph.py
@@ -100,6 +100,10 @@ class DatasetGraph:
             operation (dict): Enhancement operation.
             message (Operation): Enhancement operation proto message.
         """
+        if operation is None:
+            logger.warning("Summary can not collect the operation for dataset graph as the operation is None, "
+                           "please check the dataset.")
+            return
         for key, value in operation.items():
             if isinstance(value, (list, tuple)):
                 if all(isinstance(ele, int) for ele in value):