code: def __init__(
      self,
      inplanes: int,
      squeeze_planes: int,
      expand1x1_planes: int,
      expand3x3_planes: int,
  ) -> None:
      super().__init__()
      self.inplanes = inplanes
      self.squeeze = nn.Conv2d(inplanes, squeeze_planes, kernel_size=1)
      self.squeeze_activation = nn.ReLU(inplace=True)
      self.expand1x1 = nn.Conv2d(
          squeeze_planes, expand1x1_planes, kernel_size=1
      )
      self.expand1x1_activation = nn.ReLU(inplace=True)
      self.expand3x3 = nn.Conv2d(
          squeeze_planes, expand3x3_planes, kernel_size=3, padding=1
      )
      self.expand3x3_activation = nn.ReLU(inplace=True)
  
imports: from functools import partial
  from typing import Any, Optional
  import torch
  import torch.nn
  import torch.nn.init
  from ..transforms._presets import ImageClassification
  from ..utils import _log_api_usage_once
  from ._api import register_model, Weights, WeightsEnum
  from ._meta import _IMAGENET_CATEGORIES
  from ._utils import _ovewrite_named_param, handle_legacy_interface
label_desc: Use image node_id 1 to create a new Fire object from inherited base classes: nn with example: obj = Fire(inplanes, squeeze_planes, expand1x1_planes, expand3x3_planes)