PyTorch
Installation​
pip install superduper_torch
API​
superduper allows users to work with arbitrary torch models, with custom pre-, post-processing and input/ output data-types,
as well as offering training with superduper
| Class | Description | GitHub | API-docs |
|---|---|---|---|
superduper.ext.torch.model.TorchModel | Wraps a PyTorch model | Code | Docs |
superduper.ext.torch.model.TorchTrainer | May be attached to a TorchModel for training | Code | Docs |