hezar.trainer package

Submodules

• hezar.trainer.metrics_handlers module
  • Image2TextMetricHandler
    • Image2TextMetricHandler.compute_metrics()
    • Image2TextMetricHandler.valid_metrics
  • MetricsHandler
    • MetricsHandler.compute_metrics()
    • MetricsHandler.valid_metrics
  • SequenceLabelingMetricsHandler
    • SequenceLabelingMetricsHandler.compute_metrics()
    • SequenceLabelingMetricsHandler.valid_metrics
  • SpeechRecognitionMetricsHandler
    • SpeechRecognitionMetricsHandler.compute_metrics()
    • SpeechRecognitionMetricsHandler.valid_metrics
  • TextClassificationMetricsHandler
    • TextClassificationMetricsHandler.compute_metrics()
    • TextClassificationMetricsHandler.valid_metrics
  • TextGenerationMetricsHandler
    • TextGenerationMetricsHandler.compute_metrics()
    • TextGenerationMetricsHandler.valid_metrics
• hezar.trainer.trainer module
  • Trainer
    • Trainer.compute_loss()
    • Trainer.create_eval_dataloader()
    • Trainer.create_train_dataloader()
    • Trainer.dataset_config_file
    • Trainer.default_lr_scheduler
    • Trainer.default_optimizer
    • Trainer.evaluate()
    • Trainer.evaluation_step()
    • Trainer.forward()
    • Trainer.inner_training_loop()
    • Trainer.load_csv_logs()
    • Trainer.log()
    • Trainer.lr_scheduler_file
    • Trainer.optimization_step()
    • Trainer.optimizer_file
    • Trainer.prepare_input_batch()
    • Trainer.print_info()
    • Trainer.push_to_hub()
    • Trainer.save()
    • Trainer.train()
    • Trainer.trainer_config_file
    • Trainer.trainer_csv_log_file
    • Trainer.trainer_state_file
    • Trainer.trainer_subfolder
    • Trainer.training_step()
• hezar.trainer.trainer_utils module
  • AverageMeter
    • AverageMeter.reset()
    • AverageMeter.update()
  • CSVLogger
    • CSVLogger.write()
  • MetricsTracker
    • MetricsTracker.avg()
    • MetricsTracker.reset()
    • MetricsTracker.update()
  • TrainerState
    • TrainerState.best_checkpoint
    • TrainerState.best_metric_value
    • TrainerState.epoch
    • TrainerState.epoch_step
    • TrainerState.global_step
    • TrainerState.load()
    • TrainerState.logs_dir
    • TrainerState.loss_tracker_avg
    • TrainerState.loss_tracker_sum
    • TrainerState.metric_for_best_checkpoint
    • TrainerState.save()
    • TrainerState.total_epochs
    • TrainerState.update()
    • TrainerState.update_best_results()
  • get_distributed_logger()
  • get_lr_scheduler_type()
  • resolve_logdir()
  • write_to_tensorboard()

Module contents