ertk.pytorch.classification

PyTorch classification utilities.

pt_cross_validate(model_fn, x, y, *[, ...])
pt_train_val_test(model_fn, train_data, ...)
train_mtl_model(model, tasks, data[, ...])