revise docstr of alexnet to correct input_size (#26)

pull/30/head
Lei Yang 2020-08-12 16:14:19 +08:00 committed by GitHub
parent c905decddf
commit 408d92bcbe
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -8,7 +8,7 @@ from .base_backbone import BaseBackbone
class AlexNet(BaseBackbone):
"""`AlexNet <https://en.wikipedia.org/wiki/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.