code: def save_checkpoint(model, checkpoint_path):
      torch.save(model.state_dict(), checkpoint_path)
  
file_path: nni/examples/trials/mnist-pbt-tuner-pytorch/mnist.py
imports: import argparse
  import logging
  import os
  import nni
  import torch
  import torch.nn
  import torch.nn.functional
  import torch.optim
  from torchvision import datasets, transforms
label_desc: Use image node_id 3 for calling a global function with example usage: save_checkpoint(model, checkpoint_path) without return types