All modules for which code is available

• torchmate.callbacks.callback
• torchmate.callbacks.csv_logger
• torchmate.callbacks.early_stopper
• torchmate.callbacks.gradient_accumulator
• torchmate.callbacks.gradient_clipper
• torchmate.callbacks.model_checkpoint
• torchmate.callbacks.wandb_logger
• torchmate.callbacks.wandb_model_checkpoint
• torchmate.modules.refined_self_attention_sagan
• torchmate.modules.squeeze_and_excitation
• torchmate.trainer.trainer
• torchmate.utils.check_model_info
• torchmate.utils.color_text
• torchmate.utils.history_plotter
• torchmate.utils.progress_bar
• torchmate.utils.running_average