train_loop

Subpackages

• components
  • Submodules
    • callback_handler
      • CallbacksHandler
    • ddp_handler
      • DDPHandler
    • message_passing
      • MessageHandling
      • Message
      • MessageService
    • model_prediction_store
      • ModelPredictionStore
    • pred_collate_fns
      • append_predictions()
      • append_concat_predictions()
      • torch_cat_transf()
      • keep_list_transf()

Submodules

• train_loop
  • TrainLoop
    • TrainLoop.fit()
    • TrainLoop._train()
    • TrainLoop._calculate_batch_loss()
    • TrainLoop._backward_pass()
    • TrainLoop._optimizer_step()
    • TrainLoop._optimizer_zero_grad()
    • TrainLoop.should_execute_optimizer_update()
    • TrainLoop.auto_execute_end_of_epoch()
    • TrainLoop.auto_execute_end_of_training()
    • TrainLoop.parse_loss()
    • TrainLoop._print_save_loss()
    • TrainLoop.evaluate_loss_on_train_set()
    • TrainLoop.evaluate_loss_on_validation_set()
    • TrainLoop.evaluate_loss_on_test_set()
    • TrainLoop.evaluate_model_loss()
    • TrainLoop.predict_on_train_set()
    • TrainLoop.predict_on_validation_set()
    • TrainLoop.predict_on_test_set()
    • TrainLoop.predict_with_model()
    • TrainLoop.insert_metric_result_into_history()
    • TrainLoop.get_schedulers()
    • TrainLoop.get_num_training_steps()
    • TrainLoop.is_main_process()
    • TrainLoop.convert_loss_to_float_dict_format()
    • TrainLoop._train_dp()
    • TrainLoop._train_ddp()
    • TrainLoop._spawn_fit()
    • TrainLoop.__call__()
• train_loop_tracking
  • TrainLoopCheckpoint
  • TrainLoopEndSave
    • TrainLoopEndSave.check_if_result_packages_possible()
  • TrainLoopCheckpointEndSave