drytorch.lib

Contains library implementations.

Modules

aggregators

Module containing generic accumulator-based aggregators.

checkpoints

Module containing classes to save the model state and its optimizer state.

evaluations

Module containing classes for the evaluation of a model.

gradient_ops

Module containing gradient operations.

hooks

Module containing registry, callbacks, and hooks for a Trainer.

learning

Module containing classes with learning algorithm's specifications.

loading

Module containing classes nad utilities for batching a dateset.

models

Module containing classes for wrapping a torch module and its optimizer.

objectives

Module containing classes to create and combine loss and metrics.

runners

Module containing classes that run a model.

schedulers

Module containing schedulers for the learning rates.

training

Module containing classes for training a model.