Welcome to Torchmate’s documentation!

Contents:

• ⚙️ Installation
• 🚀 Quick Start
• 🛠️ Trainer
  • Trainer
• ⚓ Callbacks
  • Callback
  • CSVLogger
  • EarlyStopper
  • GradientAccumulator
  • GradientClipper
  • ModelCheckpoint
  • WandbLogger
  • WandbModelCheckpoint
• 🧪 Modules
  • RefinedSelfAttentionSAGAN
  • Squeeze and Excitation Modules
  • ChannelSELayer
  • SpatialSELayer
  • ChannelSpatialSELayer
  • SELayer
• 🧰 Utils
  • ProgressBar
  • RunningAverage
  • HistoryPlotter
  • COLOR
  • colorize_text
  • model_info

Indices and tables

• Index

• Module Index

• Search Page