Toggle Light / Dark / Auto color theme

Toggle table of contents sidebar

TORCHFITTER

torchfitter is a simple library to ease the training of PyTorch models. It features a class called Trainer that includes the basic functionality to fit models in a Keras-like style.

The library also provides a callbacks API that can be used to interact with the model during the training process, as well as a set of basic regularization procedures.

Additionally, you will find the Manager class which allows you to run multiple experiments for different random seeds.

Contents:

• Installation
  • Normal User
  • Developer
• Quick Start
  • Training Basics
  • Callbacks System
  • Regularization
  • Running multiple experiments
• torchfitter package
  • Subpackages
  • Submodules
• torchfitter.trainer package
  • Submodules
• torchfitter.utils module
• torchfitter.testing module
• torchfitter.manager package
  • Submodules
• torchfitter.regularization package
  • Submodules
• torchfitter.io module
• torchfitter.callbacks package
  • Submodules

Indices and tables

• Index

• Module Index

• Search Page