class_docstr: Callback for executing different checkpoint requirements.
code: def on_train_end(self, **kwargs):
      self._checkpoint.restore()
      self._checkpoint.finalize()
  
file_path: mmf/mmf/trainers/callbacks/checkpoint.py
imports: import logging
  from mmf.trainers.callbacks.base import Callback
  from mmf.utils.checkpoint import Checkpoint, consolidate_optim_state_dict
label_desc: Use image node_id 5 for calling the CheckpointCallback obj's underlying member method code with example usage: obj.on_train_end() without return types