From 20a5a770ff4de42b043c5c7028c4be35fbcd9161 Mon Sep 17 00:00:00 2001 From: tkhe Date: Fri, 24 Dec 2021 14:37:32 +0800 Subject: [PATCH] fix docs --- torchvision/models/detection/ssdlite.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/models/detection/ssdlite.py b/torchvision/models/detection/ssdlite.py index d35dfbf78d3..47ecf59f1e2 100644 --- a/torchvision/models/detection/ssdlite.py +++ b/torchvision/models/detection/ssdlite.py @@ -252,7 +252,7 @@ def ssdlite320_mobilenet_v3_large( "detections_per_img": 300, "topk_candidates": 300, # Rescale the input in a way compatible to the backbone: - # The following mean/std rescale the data from [0, 1] to [-1, -1] + # The following mean/std rescale the data from [0, 1] to [-1, 1] "image_mean": [0.5, 0.5, 0.5], "image_std": [0.5, 0.5, 0.5], }