Skip to content

learner

Learner package for model training and evaluation.

Modules:

  • predict

    A module for making predictions with PyTorch models using DataLoaders.

  • raytune_learner

    Ray Tune wrapper and trainable model classes for hyperparameter optimization.

  • raytune_parser

    Ray Tune results parser for extracting and saving best model configurations and weights.