class_docstr: This callback saves the initial model using the save method of the LightKit ConfigModule.
  This allows to load the model later without access to the hyper parameters needed to instantiate the class.
  Additionally, a dictionary of arguments is stored.
  
  Args:
      args_to_save: Contains the arguments that are stored.
code: def __init__(self, args_to_save: Dict[str, Any]) -> None:
      super().__init__()
      self.args_to_save = args_to_save
  
imports: import json
  from typing import Dict, Any
  from pytorch_lightning import Callback, Trainer, LightningModule
  from .common import get_save_dir_from_csvlogger
label_desc: Use image node_id 1 to create a new InitialSaveCallback object from inherited base classes: Callback with example: obj = InitialSaveCallback(args_to_save)