Welcome to torch-train’s documentation!

torch-train provides a wrapper around the torch.nn.Module to provide scikit-learn like fit() and predict() methods.

Contents:

• Installation
  • Dependencies
• Usage
  • Overview
  • Code
• Reference
  • Module
  • Progress
  • Variable Data Loader