From 1df9de861c4a0740b88bf4661c2f0993cc7a021f Mon Sep 17 00:00:00 2001 From: Zhiqiang Wang Date: Sat, 12 Feb 2022 11:47:53 +0800 Subject: [PATCH] Minor fixes --- yolort/models/transform.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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