diff --git a/yolort/models/transform.py b/yolort/models/transform.py
index 8eecc321..b951c417 100644
--- a/yolort/models/transform.py
+++ b/yolort/models/transform.py
@@ -162,7 +162,7 @@ def forward(
 
         Args:
             images (List[Tensor]): Images to be processed. In general the type of images is a list
-                of 3-dim `Tensors`, except for the dataloader in training and evaluation, the `images`
+                of 3-dim `Tensor`, except for the dataloader in training and evaluation, the `images`
                 will be a 4-dim `Tensor` in that case. Check out the belows link for more details:
                 https://github.com/zhiqwang/yolov5-rt-stack/pull/308#pullrequestreview-878689796
             targets (List[Dict[Tensor]], optional): ground-truth boxes present in the image for