Index

A

AdamConf (class in power_cogs.config.torch.torch_config) add_config() (in module power_cogs.config) (in module power_cogs.config.config_utils)

add_configs() (in module power_cogs.config) (in module power_cogs.config.config_utils) amsgrad (power_cogs.config.torch.torch_config.AdamConf attribute) anneal_strategy (power_cogs.config.torch.torch_config.OneCycleLRConf attribute)

B

base_lr (power_cogs.config.torch.torch_config.CyclicLRConf attribute) base_momentum (power_cogs.config.torch.torch_config.CyclicLRConf attribute) (power_cogs.config.torch.torch_config.OneCycleLRConf attribute) BaseDatasetConfig (class in power_cogs.config.base) (class in power_cogs.config.base.base_config) BaseLoggingConfig (class in power_cogs.config.base) (class in power_cogs.config.base.base_config) BaseModelConfig (class in power_cogs.config.base) (class in power_cogs.config.base.base_config) BaseTorchDataset (class in power_cogs.base) (class in power_cogs.base.base_torch_dataset)

BaseTorchModel (class in power_cogs.base) (class in power_cogs.base.base_torch_model) BaseTorchTrainer (class in power_cogs.base) (class in power_cogs.base.base_torch_trainer) BaseTrainerConfig (class in power_cogs.config.base) (class in power_cogs.config.base.base_config) batch_sampler (power_cogs.config.torch.torch_config.DataLoaderConf attribute) batch_size (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute) (power_cogs.config.torch.torch_config.DataLoaderConf attribute) betas (power_cogs.config.torch.torch_config.AdamConf attribute)

C

ChainDatasetConf (class in power_cogs.config.torch.torch_config) checkpoint_at_end (power_cogs.config.tune.tune_config.TuneConfig attribute) checkpoint_freq (power_cogs.config.tune.tune_config.TuneConfig attribute) checkpoint_interval (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute) checkpoint_path (power_cogs.config.base.base_config.BaseLoggingConfig attribute) (power_cogs.config.base.BaseLoggingConfig attribute) clear_configs() (in module power_cogs.config) (in module power_cogs.config.config_utils) collate_fn (power_cogs.config.torch.torch_config.DataLoaderConf attribute)

ConcatDatasetConf (class in power_cogs.config.torch.torch_config) cooldown (power_cogs.config.torch.torch_config.ReduceLROnPlateauConf attribute) CosineAnnealingLRConf (class in power_cogs.config.torch.torch_config) CosineAnnealingWarmRestartsConf (class in power_cogs.config.torch.torch_config) create_linear_layer() (in module power_cogs.utils.torch_utils) create_linear_network() (in module power_cogs.utils.torch_utils) create_stopper() (in module power_cogs.tune.tune_wrapper) CustomMLflowLoggerCallback (class in power_cogs.tune.tune_wrapper) cycle_momentum (power_cogs.config.torch.torch_config.CyclicLRConf attribute) (power_cogs.config.torch.torch_config.OneCycleLRConf attribute) CyclicLRConf (class in power_cogs.config.torch.torch_config)

D

dataloader_config (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute) DataLoaderConf (class in power_cogs.config.torch.torch_config) dataset (power_cogs.config.torch.torch_config.DataLoaderConf attribute) (power_cogs.config.torch.torch_config.SubsetConf attribute) dataset_config (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute) DatasetConf (class in power_cogs.config.torch.torch_config)

datasets (power_cogs.config.torch.torch_config.ChainDatasetConf attribute) (power_cogs.config.torch.torch_config.ConcatDatasetConf attribute) Config (class in power_cogs.config.base) (class in power_cogs.config.base.base_config) device_id (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute) div_factor (power_cogs.config.torch.torch_config.OneCycleLRConf attribute) doublewrap() (in module power_cogs.utils.logging) drop_last (power_cogs.config.torch.torch_config.DataLoaderConf attribute)

E

early_stoppage (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute) epochs (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute) (power_cogs.config.torch.torch_config.OneCycleLRConf attribute)

eps (power_cogs.config.torch.torch_config.AdamConf attribute) (power_cogs.config.torch.torch_config.ReduceLROnPlateauConf attribute) eta_min (power_cogs.config.torch.torch_config.CosineAnnealingLRConf attribute) (power_cogs.config.torch.torch_config.CosineAnnealingWarmRestartsConf attribute) ExponentialLRConf (class in power_cogs.config.torch.torch_config)

F

factor (power_cogs.config.torch.torch_config.ReduceLROnPlateauConf attribute) final_div_factor (power_cogs.config.torch.torch_config.OneCycleLRConf attribute)

forward() (power_cogs.model.mnist_model.MNISTModel method) from_config() (power_cogs.base.base_torch_trainer.BaseTorchTrainer class method) (power_cogs.base.BaseTorchTrainer class method)

G

gamma (power_cogs.config.torch.torch_config.CyclicLRConf attribute) (power_cogs.config.torch.torch_config.ExponentialLRConf attribute) (power_cogs.config.torch.torch_config.MultiStepLRConf attribute) (power_cogs.config.torch.torch_config.StepLRConf attribute)

generator (power_cogs.config.torch.torch_config.DataLoaderConf attribute)

H

power_cogs (module) power_cogs.base (module) power_cogs.base.base_torch_dataset (module) power_cogs.base.base_torch_model (module) power_cogs.base.base_torch_trainer (module) power_cogs.config (module) power_cogs.config.base (module) power_cogs.config.base.base_config (module) power_cogs.config.config (module) power_cogs.config.config_utils (module) power_cogs.config.load_config (module) power_cogs.config.mnist_config (module) power_cogs.config.torch (module) power_cogs.config.torch.torch_config (module)

power_cogs.config.tune (module) power_cogs.config.tune.tune_config (module) power_cogs.dataset (module) power_cogs.dataset.mnist_dataset (module) power_cogs.model (module) power_cogs.model.mnist_model (module) power_cogs.trainer (module) power_cogs.trainer.mnist_trainer (module) power_cogs.tune (module) power_cogs.tune.tune_wrapper (module) power_cogs.utils (module) power_cogs.utils.logging (module) power_cogs.utils.torch_utils (module) power_cogs.utils.utils (module)

I

indices (power_cogs.config.torch.torch_config.SubsetConf attribute)

input_dims (power_cogs.config.mnist_config.MNISTModelConfig attribute) IterableDatasetConf (class in power_cogs.config.torch.torch_config)

L

LambdaLRConf (class in power_cogs.config.torch.torch_config) last_epoch (power_cogs.config.torch.torch_config.CosineAnnealingLRConf attribute) (power_cogs.config.torch.torch_config.CosineAnnealingWarmRestartsConf attribute) (power_cogs.config.torch.torch_config.CyclicLRConf attribute) (power_cogs.config.torch.torch_config.ExponentialLRConf attribute) (power_cogs.config.torch.torch_config.LambdaLRConf attribute) (power_cogs.config.torch.torch_config.MultiStepLRConf attribute) (power_cogs.config.torch.torch_config.MultiplicativeLRConf attribute) (power_cogs.config.torch.torch_config.OneCycleLRConf attribute) (power_cogs.config.torch.torch_config.StepLRConf attribute) load() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) load_checkpoint() (power_cogs.tune.tune_wrapper.TuneTrainer method)

load_config() (in module power_cogs.config) load_model() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) log_epoch() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) log_scalar() (power_cogs.utils.logging.TensorboardLogger method) log_text() (power_cogs.utils.logging.TensorboardLogger method) logging_config (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute) loss_threshold (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute) lr (power_cogs.config.torch.torch_config.AdamConf attribute) lr_lambda (power_cogs.config.torch.torch_config.LambdaLRConf attribute) (power_cogs.config.torch.torch_config.MultiplicativeLRConf attribute)

M

make_checkpoint() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) make_trainer_defaults() (in module power_cogs.config.base) (in module power_cogs.config.base.base_config) makedirs() (in module power_cogs.utils.utils) max_lr (power_cogs.config.torch.torch_config.CyclicLRConf attribute) (power_cogs.config.torch.torch_config.OneCycleLRConf attribute) max_momentum (power_cogs.config.torch.torch_config.CyclicLRConf attribute) (power_cogs.config.torch.torch_config.OneCycleLRConf attribute) metric (power_cogs.config.tune.tune_config.TuneConfig attribute) milestones (power_cogs.config.torch.torch_config.MultiStepLRConf attribute) min_lr (power_cogs.config.torch.torch_config.ReduceLROnPlateauConf attribute) MNISTConfig (class in power_cogs.config.mnist_config)

MNISTDataset (class in power_cogs.dataset.mnist_dataset) MNISTDatasetConfig (class in power_cogs.config.mnist_config) MNISTModel (class in power_cogs.model.mnist_model) MNISTModelConfig (class in power_cogs.config.mnist_config) MNISTTrainer (class in power_cogs.trainer.mnist_trainer) MNISTTrainerConfig (class in power_cogs.config.mnist_config) mode (power_cogs.config.torch.torch_config.CyclicLRConf attribute) (power_cogs.config.torch.torch_config.ReduceLROnPlateauConf attribute) (power_cogs.config.tune.tune_config.TuneConfig attribute) model_config (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute) MultiplicativeLRConf (class in power_cogs.config.torch.torch_config) multiprocessing_context (power_cogs.config.torch.torch_config.DataLoaderConf attribute) MultiStepLRConf (class in power_cogs.config.torch.torch_config)

N

name (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute) (power_cogs.config.tune.tune_config.TuneConfig attribute) normal_std (power_cogs.config.mnist_config.MNISTModelConfig attribute)

num_samples (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute) (power_cogs.config.tune.tune_config.TuneConfig attribute) num_workers (power_cogs.config.torch.torch_config.DataLoaderConf attribute)

O

on_trial_save() (power_cogs.tune.tune_wrapper.CustomMLflowLoggerCallback method) OneCycleLRConf (class in power_cogs.config.torch.torch_config) optimizer (power_cogs.config.torch.torch_config.CosineAnnealingLRConf attribute) (power_cogs.config.torch.torch_config.CosineAnnealingWarmRestartsConf attribute) (power_cogs.config.torch.torch_config.CyclicLRConf attribute) (power_cogs.config.torch.torch_config.ExponentialLRConf attribute) (power_cogs.config.torch.torch_config.LambdaLRConf attribute) (power_cogs.config.torch.torch_config.MultiStepLRConf attribute) (power_cogs.config.torch.torch_config.MultiplicativeLRConf attribute) (power_cogs.config.torch.torch_config.OneCycleLRConf attribute) (power_cogs.config.torch.torch_config.ReduceLROnPlateauConf attribute) (power_cogs.config.torch.torch_config.StepLRConf attribute)

optimizer_config (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute) output_activation (power_cogs.config.mnist_config.MNISTModelConfig attribute) output_dims (power_cogs.config.mnist_config.MNISTModelConfig attribute)

P

params (power_cogs.config.torch.torch_config.AdamConf attribute) patience (power_cogs.config.torch.torch_config.ReduceLROnPlateauConf attribute) pct_start (power_cogs.config.torch.torch_config.OneCycleLRConf attribute) pin_memory (power_cogs.config.torch.torch_config.DataLoaderConf attribute) post_dataset_setup() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) (power_cogs.trainer.mnist_trainer.MNISTTrainer method) post_model_setup() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) post_optimizer_setup() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) post_setup() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) post_train() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method)

post_train_iter() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) pre_dataset_setup() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) pre_model_setup() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) pre_optimizer_setup() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) pre_train() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) pre_train_iter() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) pretrained_path (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute)

R

ReduceLROnPlateauConf (class in power_cogs.config.torch.torch_config)

S

safe() (in module power_cogs.utils.logging) sample() (power_cogs.base.base_torch_dataset.BaseTorchDataset method) (power_cogs.base.BaseTorchDataset method) sampler (power_cogs.config.torch.torch_config.DataLoaderConf attribute) save() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) save_checkpoint() (power_cogs.tune.tune_wrapper.TuneTrainer method) save_config() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) scale_fn (power_cogs.config.torch.torch_config.CyclicLRConf attribute) scale_mode (power_cogs.config.torch.torch_config.CyclicLRConf attribute) scheduler_config (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute) setup() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) (power_cogs.tune.tune_wrapper.TuneTrainer method) setup_config() (in module power_cogs.config) (in module power_cogs.config.config_utils) setup_dataloader() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) setup_dataset() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method)

setup_device() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) setup_logging_and_checkpoints() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) setup_model() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) setup_optimizer() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) setup_tensorboard() (power_cogs.utils.logging.TensorboardLogger method) setup_trainer() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) shuffle (power_cogs.config.torch.torch_config.DataLoaderConf attribute) step() (power_cogs.tune.tune_wrapper.TuneTrainer method) step_size (power_cogs.config.torch.torch_config.StepLRConf attribute) step_size_down (power_cogs.config.torch.torch_config.CyclicLRConf attribute) step_size_up (power_cogs.config.torch.torch_config.CyclicLRConf attribute) StepLRConf (class in power_cogs.config.torch.torch_config) steps_per_epoch (power_cogs.config.torch.torch_config.OneCycleLRConf attribute) SubsetConf (class in power_cogs.config.torch.torch_config) summary() (power_cogs.base.base_torch_model.BaseTorchModel method) (power_cogs.base.BaseTorchModel method)

T

T_0 (power_cogs.config.torch.torch_config.CosineAnnealingWarmRestartsConf attribute) T_max (power_cogs.config.torch.torch_config.CosineAnnealingLRConf attribute) T_mult (power_cogs.config.torch.torch_config.CosineAnnealingWarmRestartsConf attribute) tensorboard() (in module power_cogs.utils.logging) tensorboard_log_path (power_cogs.config.base.base_config.BaseLoggingConfig attribute) (power_cogs.config.base.BaseLoggingConfig attribute) TensorboardLogger (class in power_cogs.utils.logging) TensorDatasetConf (class in power_cogs.config.torch.torch_config) tensors (power_cogs.config.torch.torch_config.TensorDatasetConf attribute) threshold (power_cogs.config.torch.torch_config.ReduceLROnPlateauConf attribute) threshold_mode (power_cogs.config.torch.torch_config.ReduceLROnPlateauConf attribute) timeout (power_cogs.config.torch.torch_config.DataLoaderConf attribute) to_device() (power_cogs.base.base_torch_dataset.BaseTorchDataset method) (power_cogs.base.BaseTorchDataset method) (power_cogs.base.BaseTorchTrainer method) (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.dataset.mnist_dataset.MNISTDataset method)

total_steps (power_cogs.config.torch.torch_config.OneCycleLRConf attribute) train() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) train_iter() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) (power_cogs.trainer.mnist_trainer.MNISTTrainer method) trainer (power_cogs.config.base.base_config.Config attribute) (power_cogs.config.base.Config attribute) (power_cogs.config.mnist_config.MNISTConfig attribute) tune() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method) (power_cogs.tune.tune_wrapper.TuneWrapper method) tune_config (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute) TuneConfig (class in power_cogs.config.tune.tune_config) TuneTrainer (class in power_cogs.tune.tune_wrapper) TuneWrapper (class in power_cogs.tune.tune_wrapper)

U

use_cuda (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute)

use_normal_init (power_cogs.config.mnist_config.MNISTModelConfig attribute)

V

verbose (power_cogs.config.torch.torch_config.ReduceLROnPlateauConf attribute) verify_config_dict() (in module power_cogs.config.config_utils) visualize() (power_cogs.base.base_torch_trainer.BaseTorchTrainer method) (power_cogs.base.BaseTorchTrainer method)

visualize_output (power_cogs.config.base.base_config.BaseTrainerConfig attribute) (power_cogs.config.base.BaseTrainerConfig attribute)

W

weight_decay (power_cogs.config.torch.torch_config.AdamConf attribute)

worker_init_fn (power_cogs.config.torch.torch_config.DataLoaderConf attribute)

Z

zero_bias (power_cogs.config.mnist_config.MNISTModelConfig attribute)