Index

B

BLACK (torchmate.utils.COLOR attribute) BLACK_BG (torchmate.utils.COLOR attribute)

BLUE (torchmate.utils.COLOR attribute) BLUE_BG (torchmate.utils.COLOR attribute) BOLD (torchmate.utils.COLOR attribute)

C

Callback (class in torchmate.callbacks) ChannelSELayer (class in torchmate.modules) ChannelSpatialSELayer (class in torchmate.modules) COLOR (class in torchmate.utils) colorize_text() (in module torchmate.utils)

CSE (torchmate.modules.SELayer attribute) CSSE (torchmate.modules.SELayer attribute) CSVLogger (class in torchmate.callbacks) CYAN (torchmate.utils.COLOR attribute) CYAN_BG (torchmate.utils.COLOR attribute)

D

DEFAULT_BG (torchmate.utils.COLOR attribute)

E

EarlyStopper (class in torchmate.callbacks)

END (torchmate.utils.COLOR attribute) evaluate() (torchmate.trainer.Trainer method)

F

fit() (torchmate.trainer.Trainer method) format_time() (torchmate.utils.ProgressBar static method) format_to_space_capitalized() (torchmate.utils.HistoryPlotter static method)

forward() (torchmate.modules.ChannelSELayer method) (torchmate.modules.ChannelSpatialSELayer method) (torchmate.modules.RefinedSelfAttentionSAGAN method) (torchmate.modules.SpatialSELayer method)

G

GradientAccumulator (class in torchmate.callbacks) GradientClipper (class in torchmate.callbacks)

GREEN (torchmate.utils.COLOR attribute) GREEN_BG (torchmate.utils.COLOR attribute)

H

HistoryPlotter (class in torchmate.utils)

hw_flatten() (torchmate.modules.RefinedSelfAttentionSAGAN static method)

I

INVERSE (torchmate.utils.COLOR attribute)

M

MAGENTA (torchmate.utils.COLOR attribute) MAGENTA_BG (torchmate.utils.COLOR attribute) model_info() (in module torchmate.utils) ModelCheckpoint (class in torchmate.callbacks) module torchmate.callbacks torchmate.modules torchmate.modules.squeeze_and_excitation torchmate.trainer torchmate.utils

N

NONE (torchmate.modules.SELayer attribute)

O

on_backward_end() (torchmate.callbacks.Callback method) (torchmate.callbacks.GradientClipper method) on_epoch_begin() (torchmate.callbacks.Callback method) on_epoch_end() (torchmate.callbacks.Callback method) (torchmate.callbacks.CSVLogger method) (torchmate.callbacks.EarlyStopper method) (torchmate.callbacks.ModelCheckpoint method) (torchmate.callbacks.WandbLogger method) (torchmate.callbacks.WandbModelCheckpoint method) on_experiment_begin() (torchmate.callbacks.Callback method) (torchmate.callbacks.CSVLogger method) (torchmate.callbacks.GradientAccumulator method) (torchmate.callbacks.WandbLogger method) (torchmate.callbacks.WandbModelCheckpoint method)

on_experiment_end() (torchmate.callbacks.Callback method) (torchmate.callbacks.CSVLogger method) on_predict_batch_begin() (torchmate.callbacks.Callback method) on_predict_batch_end() (torchmate.callbacks.Callback method) on_predict_begin() (torchmate.callbacks.Callback method) on_predict_end() (torchmate.callbacks.Callback method) on_train_batch_begin() (torchmate.callbacks.Callback method) (torchmate.callbacks.GradientAccumulator method) on_train_batch_end() (torchmate.callbacks.Callback method) on_train_begin() (torchmate.callbacks.Callback method) on_train_end() (torchmate.callbacks.Callback method) on_val_batch_begin() (torchmate.callbacks.Callback method) on_val_batch_end() (torchmate.callbacks.Callback method) on_val_begin() (torchmate.callbacks.Callback method) on_val_end() (torchmate.callbacks.Callback method)

P

plot_all() (torchmate.utils.HistoryPlotter method) plot_loss() (torchmate.utils.HistoryPlotter method) plot_lr() (torchmate.utils.HistoryPlotter method)

plot_metrics() (torchmate.utils.HistoryPlotter method) predict() (torchmate.trainer.Trainer method) ProgressBar (class in torchmate.utils) PURPLE (torchmate.utils.COLOR attribute)

R

RED (torchmate.utils.COLOR attribute) RED_BG (torchmate.utils.COLOR attribute)

RefinedSelfAttentionSAGAN (class in torchmate.modules) RunningAverage (class in torchmate.utils)

S

SELayer (class in torchmate.modules)

SpatialSELayer (class in torchmate.modules) SSE (torchmate.modules.SELayer attribute)

T

torchmate.callbacks module torchmate.modules module torchmate.modules.squeeze_and_excitation module

torchmate.trainer module torchmate.utils module Trainer (class in torchmate.trainer)

U

UNDERLINE (torchmate.utils.COLOR attribute)

update() (torchmate.utils.ProgressBar method) (torchmate.utils.RunningAverage method)

W

WandbLogger (class in torchmate.callbacks) WandbModelCheckpoint (class in torchmate.callbacks)

WHITE (torchmate.utils.COLOR attribute) WHITE_BG (torchmate.utils.COLOR attribute)

Y

YELLOW (torchmate.utils.COLOR attribute)

YELLOW_BG (torchmate.utils.COLOR attribute)