code: def __init__(self, model_name="resnet50", framework="pytorch"):
      super().__init__(framework=framework)
      self.model_name = model_name
      self.model = torchvision.models.resnet50(pretrained=True)
  
file_path: towhee/tests/unittests/trainer/test_weights.py
imports: import unittest
  import torch
  import torchvision
  from pathlib import Path
  from towhee.trainer.utils.trainer_utils import STATE_CHECKPOINT_NAME
  from towhee.trainer.training_config import TrainingConfig
  from towhee.operator import NNOperator
  from towhee.trainer.modelcard import ModelCard
label_desc: Use image node_id 1 to create a new MockOperator object from inherited base classes: NNOperator with example: obj = MockOperator(model_name, framework)