Index

_

__contains__() (helios.core.utils.Registry method) __enter__() (helios.core.cuda.DisableCuDNNBenchmarkContext method) (helios.core.utils.ChdirContext method) __exit__() (helios.core.cuda.DisableCuDNNBenchmarkContext method) (helios.core.utils.ChdirContext method) __iter__() (helios.core.utils.Registry method) (helios.data.samplers.ResumableDistributedSampler method) (helios.data.samplers.ResumableRandomSampler method) (helios.data.samplers.ResumableSequentialSampler method)

__len__() (helios.data.samplers.ResumableRandomSampler method) (helios.data.samplers.ResumableSequentialSampler method) __str__() (helios.core.utils.Registry method)

A

AdaptiveAvgPool2d (class in helios.nn.layers.pool) add_figure() (helios.core.loggers.tensorboard.TensorboardWriter method) add_graph() (helios.core.loggers.tensorboard.TensorboardWriter method) add_hparams() (helios.core.loggers.tensorboard.TensorboardWriter method) add_image() (helios.core.loggers.tensorboard.TensorboardWriter method) add_images() (helios.core.loggers.tensorboard.TensorboardWriter method) add_pr_curve() (helios.core.loggers.tensorboard.TensorboardWriter method) add_scalar() (helios.core.loggers.tensorboard.TensorboardWriter method) add_scalars() (helios.core.loggers.tensorboard.TensorboardWriter method)

add_text() (helios.core.loggers.tensorboard.TensorboardWriter method) advance_train_phase() (helios.data.datamodule.DataModule method) all_reduce_tensors() (in module helios.core.distributed) amp_context (helios.model.model.Model property) AMPContext (class in helios.model.model) append_metadata_to_chkpt_name() (helios.model.model.Model method) append_to_banner() (helios.model.model.Model method) autocast() (helios.model.model.Model method) average_iter_time (helios.trainer.TrainingState attribute) AverageTimer (class in helios.core.utils)

B

batch_size (helios.data.datamodule.DataLoaderParams attribute) batch_to_device() (helios.model.model.Model method)

BatchPhase (class in helios.model.model) bgr2ycbcr() (in module helios.data.functional)

C

calculate_accuracy() (in module helios.metrics.functional) calculate_f1() (in module helios.metrics.functional) calculate_mae() (in module helios.metrics.functional) calculate_mae_torch() (in module helios.metrics.functional) calculate_mAP() (in module helios.metrics.functional) calculate_precision() (in module helios.metrics.functional) calculate_psnr() (in module helios.metrics.functional) calculate_psnr_torch() (in module helios.metrics.functional) calculate_recall() (in module helios.metrics.functional) calculate_rmse() (in module helios.metrics.functional) calculate_ssim() (in module helios.metrics.functional) calculate_ssim_torch() (in module helios.metrics.functional) CalculateAccuracy (class in helios.metrics.metrics) CalculateF1 (class in helios.metrics.metrics) CalculateMAE (class in helios.metrics.metrics) CalculateMAP (class in helios.metrics.metrics) CalculatePrecision (class in helios.metrics.metrics) CalculatePSNR (class in helios.metrics.metrics) CalculateRecall (class in helios.metrics.metrics) CalculateRMSE (class in helios.metrics.metrics) CalculateSSIM (class in helios.metrics.metrics) ChdirContext (class in helios.core.utils) check_pruned() (helios.plugins.optuna.plugin.OptunaPlugin method) checkpoint_sampler() (in module helios.plugins.optuna.utils) clip_gradients() (helios.model.model.Model method) close() (helios.core.loggers.base.Logger method) (helios.core.loggers.root.RootLogger method) (helios.core.loggers.tensorboard.TensorboardWriter method) (helios.core.loggers.wandb.WandbWriter method)

close_loggers() (in module helios.core.loggers) collate_fn (helios.data.datamodule.DataLoaderParams attribute) COLLATE_FN_REGISTRY (in module helios.data.datamodule) config (helios.core.loggers.wandb.WandbArgs attribute) configure_model() (helios.plugins.optuna.plugin.OptunaPlugin method) (helios.plugins.plugin.Plugin method) configure_trainer() (helios.plugins.optuna.plugin.OptunaPlugin method) (helios.plugins.plugin.Plugin method) convert_to_hwc() (in module helios.data.functional) convert_to_list() (in module helios.core.utils) CosineAnnealingRestartLR (class in helios.scheduler.schedulers) create_collate_fn() (in module helios.data.datamodule) create_dataloader() (in module helios.data.datamodule) create_dataset() (in module helios.data.datamodule) create_default_numpy_rng() (in module helios.core.rng) create_loggers() (in module helios.core.loggers) create_loss() (in module helios.losses.utils) create_metric() (in module helios.metrics.metrics) create_model() (in module helios.model.utils) create_network() (in module helios.nn.utils) create_optimizer() (in module helios.optim.utils) create_or_load_study() (in module helios.plugins.optuna.utils) create_plugin() (in module helios.plugins.plugin) create_sampler() (in module helios.data.samplers) create_scaler() (helios.model.model.Model method) create_scheduler() (in module helios.scheduler.utils) create_study_starting_from_trial() (in module helios.plugins.optuna.utils) create_transform() (in module helios.data.transforms) current_iteration (helios.trainer.TrainingState attribute)

D

DataLoaderParams (class in helios.data.datamodule) DataModule (class in helios.data.datamodule) datamodule (helios.trainer.Trainer property) Dataset (class in helios.data.datamodule) dataset (helios.data.datamodule.Dataset attribute) dataset_iter (helios.trainer.TrainingState attribute) DATASET_REGISTRY (in module helios.data.datamodule) DatasetSplit (class in helios.data.datamodule) default_init_weights() (in module helios.nn.utils) DefaultNumpyRNG (class in helios.core.rng)

device (helios.model.model.Model property) (helios.plugins.plugin.Plugin property) dict (helios.trainer.TrainingState attribute) dict() (helios.data.datamodule.Dataset method) direction (helios.plugins.optuna.utils.StudyArgs attribute) directions (helios.plugins.optuna.utils.StudyArgs attribute) DisableCuDNNBenchmarkContext (class in helios.core.cuda) DistributedInfo (class in helios.core.distributed) drop_last (helios.data.datamodule.DataLoaderParams attribute) dtype (helios.model.model.AMPContext attribute)

E

early_stop_count (helios.trainer.TrainingState attribute) EMA (class in helios.nn.swa_utils) EPOCH (helios.trainer.TrainingUnit attribute) error() (helios.core.loggers.root.RootLogger method)

eval() (helios.model.model.Model method) exception() (helios.core.loggers.root.RootLogger method) export_to_onnx() (in module helios.onnx) extra_args (helios.core.loggers.wandb.WandbArgs attribute)

F

find_last_checkpoint() (in module helios.trainer) find_pretrained_file() (in module helios.model.utils) fit() (helios.trainer.Trainer method) flush() (helios.core.loggers.base.Logger method) (helios.core.loggers.root.RootLogger method) (helios.core.loggers.tensorboard.TensorboardWriter method) (helios.core.loggers.wandb.WandbWriter method) flush_loggers() (in module helios.core.loggers) forward() (helios.data.transforms.ToImageTensor method) (helios.losses.weighted_loss.WeightedLoss method) (helios.metrics.metrics.CalculateAccuracy method) (helios.metrics.metrics.CalculateF1 method) (helios.metrics.metrics.CalculateMAE method) (helios.metrics.metrics.CalculateMAP method) (helios.metrics.metrics.CalculatePrecision method) (helios.metrics.metrics.CalculatePSNR method) (helios.metrics.metrics.CalculateRecall method) (helios.metrics.metrics.CalculateRMSE method) (helios.metrics.metrics.CalculateSSIM method) (helios.nn.layers.pool.AdaptiveAvgPool2d method) (helios.nn.swa_utils.EMA method)

from_dict() (helios.data.datamodule.DataLoaderParams class method) from_str() (helios.data.datamodule.DatasetSplit static method) (helios.trainer.TrainingUnit class method)

G

gather_into_tensor() (in module helios.core.distributed) generator (helios.core.rng.DefaultNumpyRNG property) get() (helios.core.utils.Registry method) get_average_time() (helios.core.utils.AverageTimer method) get_default_log_name() (in module helios.core.loggers.base) get_default_numpy_rng() (in module helios.core.rng) get_default_seed() (in module helios.core.rng) get_dist_info() (in module helios.core.distributed) get_env_info_str() (in module helios.core.utils) get_from_optional() (in module helios.core.utils) get_global_rank() (in module helios.core.distributed) get_local_rank() (in module helios.core.distributed) get_local_world_size() (in module helios.core.distributed) get_logger() (in module helios.core.loggers) get_logger_state_dicts() (in module helios.core.loggers)

get_lr() (helios.scheduler.schedulers.CosineAnnealingRestartLR method) (helios.scheduler.schedulers.LinearWarmupScheduler method) (helios.scheduler.schedulers.MultiStepRestartLR method) get_model_safe_types_for_load() (in module helios.model.model) get_rng_state_dict() (in module helios.core.rng) get_root_logger() (in module helios.core.loggers) get_tensorboard_writer() (in module helios.core.loggers) get_train_steps_per_epoch() (helios.data.datamodule.DataModule method) (helios.model.model.Model method) get_trainer_safe_types_for_load() (in module helios.trainer) get_wandb_writer() (in module helios.core.loggers) get_world_size() (in module helios.core.distributed) global_epoch (helios.trainer.TrainingState attribute) global_iteration (helios.trainer.TrainingState attribute) global_print() (in module helios.core.distributed) gpu_ids (helios.trainer.Trainer property)

H

have_metrics_improved() (helios.model.model.Model method) helios module helios.chkpt_migrator module helios.core module helios.core.cuda module helios.core.distributed module helios.core.loggers module helios.core.loggers.base module helios.core.loggers.root module helios.core.loggers.tensorboard module helios.core.loggers.wandb module helios.core.rng module helios.core.utils module helios.data module helios.data.datamodule module helios.data.functional module helios.data.samplers module helios.data.transforms module helios.losses module helios.losses.utils module helios.losses.weighted_loss module helios.metrics module

helios.metrics.functional module helios.metrics.metrics module helios.model module helios.model.model module helios.model.utils module helios.nn module helios.nn.layers module helios.nn.layers.pool module helios.nn.swa_utils module helios.nn.utils module helios.onnx module helios.optim module helios.optim.utils module helios.plugins module helios.plugins.optuna module helios.plugins.optuna.plugin module helios.plugins.optuna.utils module helios.plugins.plugin module helios.scheduler module helios.scheduler.schedulers module helios.scheduler.utils module helios.trainer module

I

info() (helios.core.loggers.root.RootLogger method) init_dist() (in module helios.core.distributed) is_dist_active() (in module helios.core.distributed) is_distributed (helios.data.datamodule.DataLoaderParams attribute) (helios.data.datamodule.DataModule property) (helios.model.model.Model property) (helios.plugins.plugin.Plugin property)

is_root_logger_active() (in module helios.core.loggers) is_using_torchrun() (in module helios.core.distributed) ITERATION (helios.trainer.TrainingUnit attribute)

K

keys() (helios.core.utils.Registry method)

L

LinearWarmupScheduler (class in helios.scheduler.schedulers) load_for_testing() (helios.model.model.Model method) load_if_exists (helios.plugins.optuna.utils.StudyArgs attribute) load_image() (in module helios.data.functional) load_image_pil() (in module helios.data.functional) load_rng_state_dict() (in module helios.core.rng) load_state_dict() (helios.core.loggers.base.Logger method) (helios.core.loggers.root.RootLogger method) (helios.core.loggers.tensorboard.TensorboardWriter method) (helios.core.loggers.wandb.WandbWriter method) (helios.core.rng.DefaultNumpyRNG method) (helios.data.datamodule.DataModule method) (helios.model.model.Model method) (helios.plugins.plugin.Plugin method)

load_user_state_dict() (helios.model.model.Model method) local_print() (in module helios.core.distributed) local_rank (helios.core.distributed.DistributedInfo attribute) (helios.trainer.Trainer property) local_world_size (helios.core.distributed.DistributedInfo attribute) log_file (helios.core.loggers.root.RootLogger property) Logger (class in helios.core.loggers.base) logger (helios.core.loggers.root.RootLogger property) LoggerType (class in helios.core.loggers) LOSS_REGISTRY (in module helios.losses.utils)

M

map_loc (helios.model.model.Model property) (helios.plugins.plugin.Plugin property) metrics (helios.model.model.Model property) METRICS_REGISTRY (in module helios.metrics.metrics) migrate_checkpoints_to_current_version() (in module helios.chkpt_migrator) Model (class in helios.model.model) model (helios.trainer.Trainer property) MODEL_REGISTRY (in module helios.model.utils) module helios helios.chkpt_migrator helios.core helios.core.cuda helios.core.distributed helios.core.loggers helios.core.loggers.base helios.core.loggers.root helios.core.loggers.tensorboard helios.core.loggers.wandb helios.core.rng helios.core.utils helios.data helios.data.datamodule helios.data.functional helios.data.samplers helios.data.transforms helios.losses helios.losses.utils helios.losses.weighted_loss helios.metrics helios.metrics.functional helios.metrics.metrics helios.model helios.model.model helios.model.utils helios.nn helios.nn.layers helios.nn.layers.pool helios.nn.swa_utils helios.nn.utils helios.onnx helios.optim helios.optim.utils helios.plugins helios.plugins.optuna helios.plugins.optuna.plugin helios.plugins.optuna.utils helios.plugins.plugin helios.scheduler helios.scheduler.schedulers helios.scheduler.utils helios.trainer

module (helios.nn.swa_utils.EMA property) multiprocessing_context (helios.data.datamodule.DataLoaderParams attribute) MultiStepRestartLR (class in helios.scheduler.schedulers)

N

name (helios.core.loggers.wandb.WandbArgs attribute)

NETWORK_REGISTRY (in module helios.nn.utils) num_workers (helios.data.datamodule.DataLoaderParams attribute)

O

on_testing_batch_end() (helios.model.model.Model method) on_testing_batch_start() (helios.model.model.Model method) on_testing_end() (helios.model.model.Model method) (helios.plugins.plugin.Plugin method) on_testing_start() (helios.model.model.Model method) (helios.plugins.plugin.Plugin method) on_training_batch_end() (helios.model.model.Model method) on_training_batch_start() (helios.model.model.Model method) on_training_end() (helios.model.model.Model method) (helios.plugins.optuna.plugin.OptunaPlugin method) (helios.plugins.plugin.Plugin method)

on_training_epoch_end() (helios.model.model.Model method) on_training_epoch_start() (helios.model.model.Model method) on_training_start() (helios.model.model.Model method) (helios.plugins.plugin.Plugin method) on_validation_batch_end() (helios.model.model.Model method) on_validation_batch_start() (helios.model.model.Model method) on_validation_end() (helios.model.model.Model method) (helios.plugins.plugin.Plugin method) on_validation_start() (helios.model.model.Model method) (helios.plugins.plugin.Plugin method) OPTIMIZER_REGISTRY (in module helios.optim.utils) OptunaPlugin (class in helios.plugins.optuna.plugin)

P

params (helios.data.datamodule.Dataset attribute) persistent_workers (helios.data.datamodule.DataLoaderParams attribute) pin_memory (helios.data.datamodule.DataLoaderParams attribute) pin_memory_device (helios.data.datamodule.DataLoaderParams attribute) Plugin (class in helios.plugins.plugin) plugin_id (helios.plugins.optuna.plugin.OptunaPlugin attribute) PLUGIN_REGISTRY (in module helios.plugins.plugin)

plugins (helios.trainer.Trainer property) prefetch_factor (helios.data.datamodule.DataLoaderParams attribute) prepare_data() (helios.data.datamodule.DataModule method) process_testing_batch() (helios.plugins.plugin.Plugin method) process_training_batch() (helios.plugins.plugin.Plugin method) process_validation_batch() (helios.plugins.plugin.Plugin method) project (helios.core.loggers.wandb.WandbArgs attribute) pruner (helios.plugins.optuna.utils.StudyArgs attribute)

Q

queue (helios.trainer.Trainer property)

R

random_seed (helios.data.datamodule.DataLoaderParams attribute) rank (helios.core.distributed.DistributedInfo attribute) (helios.model.model.Model property) (helios.plugins.plugin.Plugin property) (helios.trainer.Trainer property) record() (helios.core.utils.AverageTimer method) register() (helios.core.utils.Registry method) register_plugin() (helios.trainer.Trainer method) register_trainer_types_for_safe_load() (in module helios.trainer) Registry (class in helios.core.utils) report_metrics() (helios.plugins.optuna.plugin.OptunaPlugin method) requires_cuda_support() (in module helios.core.cuda)

restore_loggers() (in module helios.core.loggers) restore_sampler() (in module helios.plugins.optuna.utils) ResumableDistributedSampler (class in helios.data.samplers) ResumableRandomSampler (class in helios.data.samplers) ResumableSampler (class in helios.data.samplers) ResumableSamplerType (in module helios.data.samplers) ResumableSequentialSampler (class in helios.data.samplers) rgb2ycbcr_torch() (in module helios.data.functional) ROOT (helios.core.loggers.LoggerType attribute) RootLogger (class in helios.core.loggers.root) run_path (helios.core.loggers.tensorboard.TensorboardWriter property) running_iter (helios.trainer.TrainingState attribute)

S

safe_barrier() (in module helios.core.distributed) safe_torch_load() (in module helios.core.utils) sampler (helios.data.datamodule.DataLoaderParams attribute) (helios.plugins.optuna.utils.StudyArgs attribute) SAMPLER_REGISTRY (in module helios.data.samplers) save_name (helios.model.model.Model property) scaler (helios.model.model.AMPContext attribute) SCHEDULER_REGISTRY (in module helios.scheduler.utils) seed_rngs() (in module helios.core.rng) set() (helios.nn.swa_utils.EMA method) set_epoch() (helios.data.samplers.ResumableSampler method) setup() (helios.core.loggers.base.Logger method) (helios.core.loggers.root.RootLogger method) (helios.core.loggers.tensorboard.TensorboardWriter method) (helios.core.loggers.wandb.WandbWriter method) (helios.data.datamodule.DataModule method) (helios.model.model.Model method) (helios.plugins.optuna.plugin.OptunaPlugin method) (helios.plugins.plugin.Plugin method) setup_loggers() (in module helios.core.loggers) should_advance_dataset_phase() (helios.model.model.Model method) should_save_checkpoint() (helios.model.model.Model method) should_training_stop (helios.plugins.plugin.UniquePluginOverrides attribute) should_training_stop() (helios.model.model.Model method) (helios.plugins.optuna.plugin.OptunaPlugin method) (helios.plugins.plugin.Plugin method)

show_tensor() (in module helios.data.functional) show_tensors() (in module helios.data.functional) shuffle (helios.data.datamodule.DataLoaderParams attribute) shutdown_dist() (in module helios.core.distributed) start() (helios.core.utils.AverageTimer method) start_iter (helios.data.samplers.ResumableDistributedSampler property) (helios.data.samplers.ResumableSampler property) start_path (helios.core.utils.ChdirContext attribute) state_dict() (helios.core.loggers.base.Logger method) (helios.core.loggers.root.RootLogger method) (helios.core.loggers.tensorboard.TensorboardWriter method) (helios.core.loggers.wandb.WandbWriter method) (helios.core.rng.DefaultNumpyRNG method) (helios.data.datamodule.DataModule method) (helios.model.model.Model method) (helios.plugins.optuna.plugin.OptunaPlugin method) (helios.plugins.plugin.Plugin method) step() (helios.scheduler.schedulers.LinearWarmupScheduler method) storage (helios.plugins.optuna.utils.StudyArgs attribute) study_name (helios.plugins.optuna.utils.StudyArgs attribute) StudyArgs (class in helios.plugins.optuna.utils) suggest() (helios.plugins.optuna.plugin.OptunaPlugin method)

T

T (in module helios.core.utils) T_Any (in module helios.core.utils) target_path (helios.core.utils.ChdirContext attribute) teardown() (helios.data.datamodule.DataModule method) tensor_to_numpy() (in module helios.data.functional) TENSORBOARD (helios.core.loggers.LoggerType attribute) TensorboardWriter (class in helios.core.loggers.tensorboard) TEST (helios.data.datamodule.DatasetSplit attribute) (helios.model.model.BatchPhase attribute) test() (helios.trainer.Trainer method) test_dataloader() (helios.data.datamodule.DataModule method) test_dataset (helios.data.datamodule.DataModule property) test_exceptions (helios.trainer.Trainer property) test_step() (helios.model.model.Model method) testing_batch (helios.plugins.plugin.UniquePluginOverrides attribute) timeout (helios.data.datamodule.DataLoaderParams attribute) to_dict() (helios.data.datamodule.DataLoaderParams method) to_y_channel() (in module helios.data.functional)

ToImageTensor (class in helios.data.transforms) TRAIN (helios.data.datamodule.DatasetSplit attribute) (helios.model.model.BatchPhase attribute) train() (helios.model.model.Model method) train_dataloader() (helios.data.datamodule.DataModule method) train_dataset (helios.data.datamodule.DataModule property) train_exceptions (helios.trainer.Trainer property) train_step() (helios.model.model.Model method) trained_state_dict() (helios.model.model.Model method) Trainer (class in helios.trainer) trainer (helios.data.datamodule.DataModule property) (helios.model.model.Model property) (helios.plugins.plugin.Plugin property) training_batch (helios.plugins.plugin.UniquePluginOverrides attribute) TrainingState (class in helios.trainer) TrainingUnit (class in helios.trainer) TRANSFORM_REGISTRY (in module helios.data.transforms) trial (helios.plugins.optuna.plugin.OptunaPlugin property) types_for_safe_load() (helios.model.model.Model method)

U

unique_overrides (helios.plugins.plugin.Plugin property) UniquePluginOverrides (class in helios.plugins.plugin)

update() (helios.nn.swa_utils.EMA method) update_all_registries() (in module helios.core.utils) user_state_dict() (helios.model.model.Model method)

V

VALID (helios.data.datamodule.DatasetSplit attribute) (helios.model.model.BatchPhase attribute) valid_dataloader() (helios.data.datamodule.DataModule method)

valid_dataset (helios.data.datamodule.DataModule property) valid_step() (helios.model.model.Model method) validation_batch (helios.plugins.plugin.UniquePluginOverrides attribute) validation_cycles (helios.trainer.TrainingState attribute)

W

WANDB (helios.core.loggers.LoggerType attribute) WandbArgs (class in helios.core.loggers.wandb) WandbWriter (class in helios.core.loggers.wandb)

warning() (helios.core.loggers.root.RootLogger method) WeightedLoss (class in helios.losses.weighted_loss) world_size (helios.core.distributed.DistributedInfo attribute)