code: def test_model(model):
      model(torch.randn(2, 3, 224, 224))
  
imports: from nni.retiarii import basic_unit
  import nni.retiarii.nn.pytorch
  import warnings
  import torch
  import torch.nn
  import torch.nn.functional
  from nni.retiarii import model_wrapper
  import sys
  from pathlib import Path
label_desc: Use image node_id 5 for calling a global function with example usage: test_model(model) without return types