code: def save_checkpoint(
      model, current_step, epoch, output_path, **kwargs
  ):
      """Save TF Vocoder model"""
      state = {
          "model": model.weights,
          "step": current_step,
          "epoch": epoch,
          "date": datetime.date.today().strftime("%B %d, %Y"),
      }
      state.update(kwargs)
      pickle.dump(state, open(output_path, "wb"))
  
imports: import datetime
  import pickle
  import tensorflow
label_desc: Use image node_id 1 for calling a global function with example usage: save_checkpoint(model, current_step, epoch, output_path) without return types