code: def save_checkpoint(
      checkpoint, model, optimizer, lr_scheduler, epoch
  ):
      if checkpoint is None:
          return
  
      state = {
          "model": model.state_dict(),
          "optimizer": optimizer.state_dict(),
          "lr_scheduler": lr_scheduler.state_dict(),
          "epoch": epoch,
      }
      torch.save(state, f"{checkpoint}/checkpoint.pth")
  
file_path: orion/examples/tutorials/code_2_hyperband_checkpoint.py
imports: import os
  import numpy
  import torch
  import torch.nn
  import torch.optim
  import torchvision
  import torchvision.models
  import torchvision.transforms
  from torch.utils.data import SubsetRandomSampler
  from orion.client import build_experiment
label_desc: Use image node_id 2 for calling a global function with example usage: save_checkpoint(checkpoint, model, optimizer, lr_scheduler, epoch) without return types