class_docstr: to save the template model, not the multi-GPU model
      
class_docstr_tok: ["to","save","the","template","model",",","not","the","multi-GPU","model"]
code: def __init__(self, model_to_save, **kwargs):
      super(CustomModelCheckpoint, self).__init__(**kwargs)
      self.model_to_save = model_to_save
  
file_path: ImageAI/imageai_tf_deprecated/Detection/Custom/callbacks.py
imports: from tensorflow.keras.callbacks import TensorBoard, ModelCheckpoint
  import tensorflow
  import numpy
  import warnings
label_desc: Use image node_id 1 to create a new CustomModelCheckpoint object from inherited base classes: ModelCheckpoint with example: obj = CustomModelCheckpoint(model_to_save)