Minor fixes

zhiqwang · Feb 12, 2022 · 1df9de8 · 1df9de8
1 parent c2fb994
commit 1df9de8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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