class_docstr: PyTorch model operator base
class_docstr_tok: ["PyTorch","model","operator","base"]
code: def __init__(self, model_name) -> None:
      super().__init__()
      model_func = getattr(torchvision.models, model_name)
      self._model = model_func(pretrained=True)
      self._model.eval()
  
file_path: towhee/tests/unittests/mock_operators/pytorch_image_classification_operator/pytorch_image_classification_operator.py
imports: import json
  import numpy
  import torch
  from typing import NamedTuple
  from pathlib import Path
  import torchvision
  from towhee.operator import Operator
label_desc: Use image node_id 1 to create a new PyTorchImageClassificationOperator object from inherited base classes: Operator with example: obj = PyTorchImageClassificationOperator(model_name)