code: def create_pytorch():
      import torch
      from pytorch_model import Net
  
      x = torch.randn(1, 3, 224, 224)
      m = Net()
      y = m(x)
      m.eval()
      torch.save(m, "test_model.pt")
  
label_desc: Use image node_id 2 for calling a global function with example usage: create_pytorch() without return types