code: def __init__(self, feature_layers=[0, 3, 5], use_normalization=False):
      super().__init__()
      # model = models.vgg19(pretrained=True)
      model = models.squeezenet1_1(pretrained=True)
      model.float()
      model.eval()
  
      self.model = model
      self.feature_layers = feature_layers
  
      self.mean = torch.FloatTensor([0.485, 0.456, 0.406])
      self.mean_tensor = None
  
      self.std = torch.FloatTensor([0.229, 0.224, 0.225])
      self.std_tensor = None
  
      self.use_normalization = use_normalization
  
      for param in self.parameters():
          param.requires_grad = False
  
file_path: BigDL/python/nano/client-application/VideoStylization/feature_model.py
imports: import torch
  import torch.nn
  from torch.autograd import Variable
  from torchvision import models
label_desc: Use image node_id 1 to create a new FeaturesNet object from inherited base classes: nn with example: obj = FeaturesNet(feature_layers, use_normalization)