Torchmate
latest
Contents:
โ๏ธ Installation
๐ Quick Start
๐ ๏ธ Trainer
โ Callbacks
๐งช Modules
๐งฐ Utils
Torchmate
Index
Edit on GitHub
Index
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
W
|
Y
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)