From 408d92bcbe5f83c296e31ca0128b3d6521da7055 Mon Sep 17 00:00:00 2001 From: Lei Yang Date: Wed, 12 Aug 2020 16:14:19 +0800 Subject: [PATCH] revise docstr of alexnet to correct input_size (#26) --- mmcls/models/backbones/alexnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmcls/models/backbones/alexnet.py b/mmcls/models/backbones/alexnet.py index 541cd547..ee1d2afe 100644 --- a/mmcls/models/backbones/alexnet.py +++ b/mmcls/models/backbones/alexnet.py @@ -8,7 +8,7 @@ from .base_backbone import BaseBackbone class AlexNet(BaseBackbone): """`AlexNet `_ backbone. - The input for AlexNet is a 256x256 RGB image. + The input for AlexNet is a 224x224 RGB image. Args: num_classes (int): number of classes for classification.