code: def save_pt_file(filepath="base_model.pt"):
      model = ArgmaxModel()
      torch.save(model.state_dict(), filepath)
  
imports: import torch
label_desc: Use image node_id 1 for calling a global function with example usage: save_pt_file(filepath) without return types