code: def __init__(self, torch_hub: Optional[str] = None) -> None:
      super().__init__()
      if torch_hub is None:
          torch_hub = "facebookresearch/swav:main"
  
      self.model = torch.hub.load(torch_hub, "resnet50")
      self.model.fc = nn.Identity()
      # put into evaluation mode
      self.eval()
  
file_path: NeuralCompression/neuralcompression/metrics/_fid_swav.py
imports: from typing import Any, List, Optional
  import torch
  import torch.nn
  from torch import Tensor
  from torch.nn import Module
  from torch_fidelity.metric_fid import fid_features_to_statistics, fid_statistics_to_metric
  from torchmetrics.metric import Metric
  from torchvision.transforms import Normalize
label_desc: Use image node_id 1 to create a new NoTrainSwAV object from inherited base classes: Module with example: obj = NoTrainSwAV(torch_hub)