Skip to main content

PyTorch

superduperdb allows users to work with arbitrary torch models, with custom pre-, post-processing and input/ output data-types, as well as offering training with superduperdb

ClassDescriptionGitHubAPI-docs
superduperdb.ext.torch.model.TorchModelWraps a PyTorch modelCode...
superduperdb.ext.torch.model.TorchTrainerMay be attached to a TorchModel for trainingCode...