PyTorch backend

This package contains the PyTorch backend for the ERTK.

models

PyTorch implementations of various models.

train

Training configuration for PyTorch models.

dataset

PyTorch dataset utilities.

classification

PyTorch classification utilities.

utils

PyTorch utilities.