squeezenet.rst 734 B

1234567891011121314151617181920212223242526
  1. SqueezeNet
  2. ==========
  3. .. currentmodule:: torchvision.models
  4. The SqueezeNet model is based on the `SqueezeNet: AlexNet-level accuracy with
  5. 50x fewer parameters and <0.5MB model size <https://arxiv.org/abs/1602.07360>`__
  6. paper.
  7. Model builders
  8. --------------
  9. The following model builders can be used to instantiate a SqueezeNet model, with or
  10. without pre-trained weights. All the model builders internally rely on the
  11. ``torchvision.models.squeezenet.SqueezeNet`` base class. Please refer to the `source
  12. code
  13. <https://github.com/pytorch/vision/blob/main/torchvision/models/squeezenet.py>`_ for
  14. more details about this class.
  15. .. autosummary::
  16. :toctree: generated/
  17. :template: function.rst
  18. squeezenet1_0
  19. squeezenet1_1