torch-train

Training wrapper around torch.nn.Module, provides scikit-learn like fit and predict interfaces.

Installation

The easiest way to install is through pip

pip install torch-train

Dependencies

The torch-train is an extension of the pytorch library. Therefore, when installed manually, ensure pytorch is installed. Either through pip:

pip install torch

See https://pytorch.org/ for a detailed installation guide.

Documentation

For a full reference, please see torch-train.readthedocs.io.