dataeval.utils.training

Utility functions for training and inference with PyTorch models.

Functions

predict(x, model[, device, batch_size, preprocess_fn])

Make batch predictions on a model.

train(model, x_train, y_train, loss_fn, optimizer, ...)

Train PyTorch model.