train_loop
Subpackages
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