diff --git a/paddlex/cv/models/yolo_v3.py b/paddlex/cv/models/yolo_v3.py
index 89c3bbc75e15cdd7b1af4839798e33e114107c5f..b87b6679a6b0d38b922912c54ec4b98d1d6b1c4e 100644
--- a/paddlex/cv/models/yolo_v3.py
+++ b/paddlex/cv/models/yolo_v3.py
@@ -266,7 +266,7 @@ class YOLOv3(BaseAPI):
                     if isinstance(bt, paddlex.det.transforms.BatchRandomShape):
                         self.max_shape = max(bt.random_shapes)
                         break
-            self.init_params['max_shape'] = max_shape
+            self.init_params['max_shape'] = self.max_shape
             iou_bt = paddlex.det.transforms.GenerateYoloTarget
             train_dataset.transforms.batch_transforms.append(iou_bt(anchors=self.anchors,
                                                                   anchor_masks=self.anchor_masks,