Model Core Utils

core classes & helpers

Classifier Abstract Base Class

Classifier

 Classifier (num_classes:int, optimizer:torch.optim.optimizer.Optimizer, s
             cheduler:<module'torch.optim.lr_scheduler'from'/opt/hostedtoo
             lcache/Python/3.10.16/x64/lib/python3.10/site-
             packages/torch/optim/lr_scheduler.py'>)

Helper class that provides a standard way to create an ABC using inheritance.

source

plot_classifier_metrics_from_csv

 plot_classifier_metrics_from_csv (metrics_csv_path:str|os.PathLike)

Sequential Model

source

SequentialModelX

 SequentialModelX (modules:List[torch.nn.modules.module.Module], *args,
                   **kwargs)

Helper class that provides a standard way to create an ABC using inheritance.

LR Finder Helper

source

find_optimal_lr

 find_optimal_lr (model, train_loader, criterion=None, optimizer=None,
                  device='cuda')