:github_url: Heimdall.trainer.HeimdallTrainer

Heimdall.trainer.HeimdallTrainer
================================

.. currentmodule:: Heimdall.trainer

.. autoclass:: HeimdallTrainer

   
   
   .. rubric:: Attributes

   .. autosummary::
      :toctree: .
   
       ~HeimdallTrainer.CHECKPOINT_KEYS
       ~HeimdallTrainer.data
       ~HeimdallTrainer.dataset_cfg
       ~HeimdallTrainer.fc_cfg
       ~HeimdallTrainer.fe_cfg
       ~HeimdallTrainer.fg_cfg
       ~HeimdallTrainer.get_precomputed
       ~HeimdallTrainer.local_cfg
       ~HeimdallTrainer.model_cfg
       ~HeimdallTrainer.optimizer_cfg
       ~HeimdallTrainer.save_precomputed
       ~HeimdallTrainer.scheduler_cfg
       ~HeimdallTrainer.trainer_cfg
   
   

   
   
   .. rubric:: Methods

   .. autosummary::
      :toctree: .
   
          ~HeimdallTrainer.check_flash_attn
          ~HeimdallTrainer.fit
          ~HeimdallTrainer.fit_model
          ~HeimdallTrainer.get_checkpoint_directory
          ~HeimdallTrainer.get_latest_checkpoint_path
          ~HeimdallTrainer.get_outputs_and_loss
          ~HeimdallTrainer.get_precomputed_outputs
          ~HeimdallTrainer.get_pretrained_load_path
          ~HeimdallTrainer.initialize_checkpointing
          ~HeimdallTrainer.instantiate_loss_functions_from_config
          ~HeimdallTrainer.iterate_dataloader
          ~HeimdallTrainer.load_checkpoint
          ~HeimdallTrainer.load_pretrained
          ~HeimdallTrainer.load_trainer_state
          ~HeimdallTrainer.print_r0
          ~HeimdallTrainer.save_checkpoint
          ~HeimdallTrainer.save_precomputed_outputs
          ~HeimdallTrainer.save_umaps
          ~HeimdallTrainer.setup_class_names_and_num_labels
          ~HeimdallTrainer.train_epoch
          ~HeimdallTrainer.validate_model