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.
torch-train provides a wrapper around the torch.nn.Module to provide scikit-learn like fit()
and predict()
methods.