SqueezeNet¶
-
class
SqueezeNet(version, classes=1000, **kwargs)¶ SqueezeNet model from the “SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size” paper. SqueezeNet 1.1 model from the official SqueezeNet repo. SqueezeNet 1.1 has 2.4x less computation and slightly fewer parameters than SqueezeNet 1.0, without sacrificing accuracy.
Parameters: - version : str
Version of squeezenet. Options are ‘1.0’, ‘1.1’.
- classes : int, default 1000
Number of classification classes.