Index

A

add_image() (torchpack.callbacks.writers.SummaryWriter method) (torchpack.train.summary.Summary method) add_scalar() (torchpack.callbacks.writers.SummaryWriter method) (torchpack.train.summary.Summary method) after_epoch() (torchpack.train.trainer.Trainer method)

after_step() (torchpack.train.trainer.Trainer method) after_train() (torchpack.train.trainer.Trainer method) allgather() (in module torchpack.distributed.comm) allreduce() (in module torchpack.distributed.comm) auto_set_run_dir() (in module torchpack.environ.rundir)

B

barrier() (in module torchpack.distributed.comm) batched_index_select() (in module torchpack.nn.functional.index)

before_epoch() (torchpack.train.trainer.Trainer method) before_step() (torchpack.train.trainer.Trainer method) before_train() (torchpack.train.trainer.Trainer method)

C

CategoricalAccuracy (class in torchpack.callbacks.metrics) CIFAR (class in torchpack.datasets.vision.cifar)

Config (class in torchpack.utils.config) ConsoleWriter (class in torchpack.callbacks.writers)

D

Dataset (class in torchpack.datasets.dataset)

dict() (torchpack.utils.config.Config method)

E

EnableCallbackIf (class in torchpack.callbacks.triggers) EstimatedTimeLeft (class in torchpack.callbacks.progress)

extreme (torchpack.callbacks.checkpoint.MaxSaver attribute) (torchpack.callbacks.checkpoint.MinSaver attribute)

F

forward() (torchpack.models.vision.mobilenetv1.MobileNetV1 method) (torchpack.models.vision.mobilenetv2.MobileBlockV2 method) (torchpack.models.vision.mobilenetv2.MobileNetV2 method) (torchpack.models.vision.shufflenetv2.ShuffleBlockV2 method) (torchpack.models.vision.shufflenetv2.ShuffleNetV2 method)

G

get_commit_hash() (in module torchpack.utils.git) get_remote_url() (in module torchpack.utils.git)

get_run_dir() (in module torchpack.environ.rundir) GPUUtilizationTracker (class in torchpack.callbacks.trackers)

H

hash() (torchpack.utils.config.Config method)

I

ImageNet (class in torchpack.datasets.vision.imagenet) InferenceRunner (class in torchpack.callbacks.inference) init() (in module torchpack.distributed.context)

is_inside_work_tree() (in module torchpack.utils.git) is_master() (in module torchpack.distributed.context) items() (torchpack.train.summary.Summary method)

J

JSONLWriter (class in torchpack.callbacks.writers)

K

keys() (torchpack.train.summary.Summary method)

L

layers (torchpack.models.vision.mobilenetv1.MobileNetV1 attribute) (torchpack.models.vision.mobilenetv2.MobileNetV2 attribute) (torchpack.models.vision.shufflenetv2.ShuffleNetV2 attribute) load() (in module torchpack.utils.io) (torchpack.utils.config.Config method) load_json() (in module torchpack.utils.io) load_jsonl() (in module torchpack.utils.io) load_mat() (in module torchpack.utils.io)

load_npy() (in module torchpack.utils.io) load_npz() (in module torchpack.utils.io) load_pt() (in module torchpack.utils.io) load_source() (in module torchpack.utils.imp) load_state_dict() (torchpack.train.trainer.Trainer method) load_yaml() (in module torchpack.utils.io) local_rank() (in module torchpack.distributed.context) local_size() (in module torchpack.distributed.context)

M

main() (in module torchpack.launch.assets.silentrun) (in module torchpack.launch.launchers.drunner) (in module torchpack.launch.main) make_divisible() (in module torchpack.models.utils) makedir() (in module torchpack.utils.fs) master_only (torchpack.callbacks.checkpoint.Saver attribute) (torchpack.callbacks.metainfo.MetaInfoSaver attribute) (torchpack.callbacks.progress.EstimatedTimeLeft attribute) (torchpack.callbacks.progress.ProgressBar attribute) (torchpack.callbacks.trackers.GPUUtilizationTracker attribute) (torchpack.callbacks.trackers.ThroughputTracker attribute) (torchpack.callbacks.writers.SummaryWriter attribute)

match() (torchpack.utils.matching.NameMatcher method) MaxSaver (class in torchpack.callbacks.checkpoint) MeanAbsoluteError (class in torchpack.callbacks.metrics) MeanSquaredError (class in torchpack.callbacks.metrics) MetaInfoSaver (class in torchpack.callbacks.metainfo) MinSaver (class in torchpack.callbacks.checkpoint) MobileBlockV1 (class in torchpack.models.vision.mobilenetv1) MobileBlockV2 (class in torchpack.models.vision.mobilenetv2) MobileNetV1 (class in torchpack.models.vision.mobilenetv1) MobileNetV2 (class in torchpack.models.vision.mobilenetv2)

N

NameMatcher (class in torchpack.utils.matching)

naturaldelta() (in module torchpack.utils.humanize) normpath() (in module torchpack.utils.fs)

P

parse_cuda_devices() (in module torchpack.utils.device) PeriodicCallback (class in torchpack.callbacks.triggers)

PeriodicTrigger (class in torchpack.callbacks.triggers) ProgressBar (class in torchpack.callbacks.progress)

R

rank() (in module torchpack.distributed.context) reload() (torchpack.utils.config.Config method) remove() (in module torchpack.utils.fs)

reset_parameters() (torchpack.models.vision.mobilenetv1.MobileNetV1 method) (torchpack.models.vision.mobilenetv2.MobileNetV2 method) (torchpack.models.vision.shufflenetv2.ShuffleNetV2 method) run_step() (torchpack.train.trainer.Trainer method)

S

save() (in module torchpack.utils.io) save_json() (in module torchpack.utils.io) save_jsonl() (in module torchpack.utils.io) save_mat() (in module torchpack.utils.io) save_npy() (in module torchpack.utils.io) save_npz() (in module torchpack.utils.io) save_pt() (in module torchpack.utils.io) save_yaml() (in module torchpack.utils.io) Saver (class in torchpack.callbacks.checkpoint) SaverRestore (class in torchpack.callbacks.checkpoint)

set_cuda_visible_devices() (in module torchpack.utils.device) set_run_dir() (in module torchpack.environ.rundir) set_trainer() (torchpack.train.summary.Summary method) ShuffleBlockV2 (class in torchpack.models.vision.shufflenetv2) ShuffleNetV2 (class in torchpack.models.vision.shufflenetv2) size() (in module torchpack.distributed.context) state_dict() (torchpack.train.trainer.Trainer method) StopTraining Summary (class in torchpack.train.summary) SummaryWriter (class in torchpack.callbacks.writers)

T

TFEventWriter (class in torchpack.callbacks.writers) ThroughputTracker (class in torchpack.callbacks.trackers) TopKCategoricalAccuracy (class in torchpack.callbacks.metrics) torchpack (module) torchpack.callbacks (module) torchpack.callbacks.checkpoint (module) torchpack.callbacks.inference (module) torchpack.callbacks.metainfo (module) torchpack.callbacks.metrics (module) torchpack.callbacks.progress (module) torchpack.callbacks.trackers (module) torchpack.callbacks.triggers (module) torchpack.callbacks.writers (module) torchpack.datasets (module) torchpack.datasets.dataset (module) torchpack.datasets.vision (module) torchpack.datasets.vision.cifar (module) torchpack.datasets.vision.imagenet (module) torchpack.distributed (module) torchpack.distributed.comm (module) torchpack.distributed.context (module) torchpack.environ (module) torchpack.environ.rundir (module) torchpack.launch (module) torchpack.launch.assets (module) torchpack.launch.assets.silentrun (module) torchpack.launch.launchers (module) torchpack.launch.launchers.drunner (module) torchpack.launch.main (module)

torchpack.models (module) torchpack.models.utils (module) torchpack.models.vision (module) torchpack.models.vision.mobilenetv1 (module) torchpack.models.vision.mobilenetv2 (module) torchpack.models.vision.shufflenetv2 (module) torchpack.nn (module) torchpack.nn.functional (module) torchpack.nn.functional.index (module) torchpack.train (module) torchpack.train.exception (module) torchpack.train.summary (module) torchpack.train.trainer (module) torchpack.utils (module) torchpack.utils.config (module) torchpack.utils.device (module) torchpack.utils.fs (module) torchpack.utils.git (module) torchpack.utils.humanize (module) torchpack.utils.imp (module) torchpack.utils.io (module) torchpack.utils.logging (module) torchpack.utils.matching (module) torchpack.utils.typing (module) torchpack.version (module) train() (torchpack.train.trainer.Trainer method) train_with_defaults() (torchpack.train.trainer.Trainer method) Trainer (class in torchpack.train.trainer) trigger_epoch() (torchpack.train.trainer.Trainer method) trigger_step() (torchpack.train.trainer.Trainer method)

U

update (torchpack.utils.config.Config attribute)

V

values() (torchpack.train.summary.Summary method)