banhxeo.train package

Submodules

• banhxeo.train.callbacks module
  • TrainerCallback
    • TrainerCallback.name
    • TrainerCallback.get_output()
    • TrainerCallback.on_init_end()
    • TrainerCallback.on_train_begin()
    • TrainerCallback.on_train_end()
    • TrainerCallback.on_epoch_begin()
    • TrainerCallback.on_epoch_end()
    • TrainerCallback.on_step_begin()
    • TrainerCallback.on_step_end()
    • TrainerCallback.on_evaluate()
    • TrainerCallback.on_save()
  • ProgressCallback
    • ProgressCallback.name
    • ProgressCallback.__init__()
    • ProgressCallback.on_train_begin()
    • ProgressCallback.on_epoch_begin()
    • ProgressCallback.on_train_end()
    • ProgressCallback.on_step_end()
  • AccuracyCallback
    • AccuracyCallback.name
    • AccuracyCallback.__init__()
    • AccuracyCallback.on_step_end()
    • AccuracyCallback.on_evaluate()
    • AccuracyCallback.get_output()
  • CheckpointCallback
    • CheckpointCallback.name
    • CheckpointCallback.__init__()
    • CheckpointCallback.on_step_end()
    • CheckpointCallback.on_epoch_end()
• banhxeo.train.config module
  • LossConfig
    • LossConfig.name
    • LossConfig.kwargs
    • LossConfig.get_loss_function()
    • LossConfig.model_config
  • OptimizerConfig
    • OptimizerConfig.name
    • OptimizerConfig.scheduler_name
    • OptimizerConfig.optimizer_kwargs
    • OptimizerConfig.lr_scheduler_kwargs
    • OptimizerConfig.Config
      • OptimizerConfig.Config.arbitrary_types_allowed
    • OptimizerConfig.get_optimizer()
    • OptimizerConfig.get_scheduler()
    • OptimizerConfig.model_config
  • TrainerConfig
    • TrainerConfig.output_dir
    • TrainerConfig.num_train_epochs
    • TrainerConfig.per_device_train_batch_size
    • TrainerConfig.per_device_eval_batch_size
    • TrainerConfig.gradient_accumulation_steps
    • TrainerConfig.training_shuffle
    • TrainerConfig.logging_steps
    • TrainerConfig.save_steps
    • TrainerConfig.save_total_limit
    • TrainerConfig.evaluate_during_training
    • TrainerConfig.evaluation_steps
    • TrainerConfig.seed
    • TrainerConfig.optim
    • TrainerConfig.loss
    • TrainerConfig.check_grad_acc_steps()
    • TrainerConfig.model_config
• banhxeo.train.trainer module
  • Trainer
    • Trainer.__init__()
    • Trainer.get_callbacks_output()
    • Trainer.get_train_dataloader()
    • Trainer.get_eval_dataloader()
    • Trainer.train()
    • Trainer.evaluate()
    • Trainer.save_model()