torchtrain

aitoolbox.torchtrain is the main user-facing API of the AIToolbox package. It incorporates the PyTorch model training via the train loop engine as well as automatic experiment progress and performance tracking. The experiment results are either stored only locally or if desired also automatically synced to the selected cloud storage (AWS S3 or Google Cloud Storage).

Guides:

• Train Loop
• TTModel
• Callbacks
• Schedulers
• Multi-Loss and Multi-Optimizer
• Multi-GPU Training
• Automatic Mixed Precision Training
• Advanced Topics