Index

A

apply_on_batch() (DenseDepth static method)

B

BaseTrainer (class in vathos.trainer.base_trainer) BCEDiceLoss (class in vathos.model.loss)

BCETverskyLoss (class in vathos.model.loss) BerHuLoss (class in vathos.model.loss)

D

DenseDepth (class in vathos.data_loader)

DiceLoss (class in vathos.model.loss)

E

extractall() (DenseDepth method)

G

get_instance() (in module vathos.utils) get_instance_v2() (in module vathos.utils)

GPUTrainer (class in vathos.trainer) GradLoss (class in vathos.model.loss)

I

iou() (in module vathos.model.loss)

L

load_config() (in module vathos.utils)

O

optimizer_to() (in module vathos.trainer.base_trainer)

P

plot4_batch() (DenseDepth static method)

plot_results() (DenseDepth static method) plot_sample() (DenseDepth static method)

R

ResDoubleConv (class in vathos.model.resunet_v2) (class in vathos.model.resunext_v2) ResDownBlock (class in vathos.model.resunet_v2) (class in vathos.model.resunext_v2) ResUNet (class in vathos.model.resunet_v2) ResUNeXt (class in vathos.model.resunext_v2)

ResUpBlock (class in vathos.model.resunet_v2) (class in vathos.model.resunext_v2) rmse() (in module vathos.model.loss) RMSELoss (class in vathos.model.loss) RMSEwSSIMLoss (class in vathos.model.loss) Runner (class in vathos.runner)

S

scheduler_to() (in module vathos.trainer.base_trainer) setup_device() (in module vathos.utils) setup_logger() (in module vathos.utils) setup_param_groups() (in module vathos.utils)

setup_train() (Runner method) SSIMLoss (class in vathos.model.loss) start_train() (GPUTrainer method) (Runner method)

T

test_epoch() (GPUTrainer method) TPUTrainer (class in vathos.trainer)

train_epoch() (GPUTrainer method) TverskyLoss (class in vathos.model.loss)