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.