Index

_

__init__() (torchkit.checkpoint.Checkpoint method) (torchkit.checkpoint.CheckpointManager method) (torchkit.layers.CausalConv1d method) (torchkit.layers.Flatten method) (torchkit.layers.GlobalAvgPool1d method) (torchkit.layers.GlobalAvgPool2d method) (torchkit.layers.GlobalAvgPool3d method) (torchkit.layers.GlobalMaxPool1d method) (torchkit.layers.GlobalMaxPool2d method) (torchkit.layers.GlobalMaxPool3d method) (torchkit.layers.SpatialSoftArgmax method) (torchkit.logger.Logger method) (torchkit.utils.timer.Stopwatch method)

B

bias (torchkit.layers.CausalConv1d attribute)

C

CausalConv1d (class in torchkit.layers) Checkpoint (class in torchkit.checkpoint) CheckpointManager (class in torchkit.checkpoint) close() (torchkit.logger.Logger method)

conv2d() (in module torchkit.layers) conv3d() (in module torchkit.layers) copy_config_and_replace() (in module torchkit.utils.config) cross_entropy() (in module torchkit.losses)

D

dilation (torchkit.layers.CausalConv1d attribute)

dump_config() (in module torchkit.utils.config)

E

elapsed() (torchkit.utils.timer.Stopwatch method)

F

Flatten (class in torchkit.layers)

flush() (torchkit.logger.Logger method)

G

get_total_params() (in module torchkit.utils.module_stats) git_commit_hash() (in module torchkit.utils.git) git_revision_hash() (in module torchkit.utils.git) GlobalAvgPool1d (class in torchkit.layers) GlobalAvgPool2d (class in torchkit.layers)

GlobalAvgPool3d (class in torchkit.layers) GlobalMaxPool1d (class in torchkit.layers) GlobalMaxPool2d (class in torchkit.layers) GlobalMaxPool3d (class in torchkit.layers) groups (torchkit.layers.CausalConv1d attribute)

H

huber_loss() (in module torchkit.losses)

I

infinite_dataset() (in module torchkit.utils.dataset)

K

kernel_size (torchkit.layers.CausalConv1d attribute)

L

latest_checkpoint (torchkit.checkpoint.CheckpointManager property) list_checkpoints() (torchkit.checkpoint.CheckpointManager static method) load_checkpoint_at() (torchkit.checkpoint.CheckpointManager method) load_config() (in module torchkit.utils.config) load_latest_checkpoint() (torchkit.checkpoint.CheckpointManager method)

load_pickle() (in module torchkit.utils.io) log_image() (torchkit.logger.Logger method) log_learning_rate() (torchkit.logger.Logger method) log_scalar() (torchkit.logger.Logger method) log_video() (torchkit.logger.Logger method) Logger (class in torchkit.logger)

M

module torchkit.experiment torchkit.layers torchkit.losses torchkit.utils.config torchkit.utils.dataset torchkit.utils.git torchkit.utils.io torchkit.utils.module_stats torchkit.utils.pdb_fallback torchkit.utils.seed torchkit.utils.timer

O

one_hot() (in module torchkit.losses)

out_channels (torchkit.layers.CausalConv1d attribute) output_padding (torchkit.layers.CausalConv1d attribute)

P

padding (torchkit.layers.CausalConv1d attribute)

padding_mode (torchkit.layers.CausalConv1d attribute) pdb_fallback() (in module torchkit.utils.pdb_fallback)

R

reset() (torchkit.utils.timer.Stopwatch method)

restore() (torchkit.checkpoint.Checkpoint method) restore_or_initialize() (torchkit.checkpoint.CheckpointManager method)

S

save() (torchkit.checkpoint.Checkpoint method) (torchkit.checkpoint.CheckpointManager method) save_pickle() (in module torchkit.utils.io) seed_rngs() (in module torchkit.utils.seed) set_cudnn() (in module torchkit.utils.seed)

setup_experiment() (in module torchkit.experiment) SpatialSoftArgmax (class in torchkit.layers) Stopwatch (class in torchkit.utils.timer) stride (torchkit.layers.CausalConv1d attribute) string_from_kwargs() (in module torchkit.experiment)

T

torchkit.experiment module torchkit.layers module torchkit.losses module torchkit.utils.config module torchkit.utils.dataset module torchkit.utils.git module torchkit.utils.io module torchkit.utils.module_stats module

torchkit.utils.pdb_fallback module torchkit.utils.seed module torchkit.utils.timer module training (torchkit.layers.Flatten attribute) (torchkit.layers.GlobalAvgPool1d attribute) (torchkit.layers.GlobalAvgPool2d attribute) (torchkit.layers.GlobalAvgPool3d attribute) (torchkit.layers.GlobalMaxPool1d attribute) (torchkit.layers.GlobalMaxPool2d attribute) (torchkit.layers.GlobalMaxPool3d attribute) (torchkit.layers.SpatialSoftArgmax attribute) transposed (torchkit.layers.CausalConv1d attribute)

U

unique_id() (in module torchkit.experiment)

V

validate_config() (in module torchkit.utils.config)

W

weight (torchkit.layers.CausalConv1d attribute)