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.logging.TensorboardWriter method) add_graph() (helios.core.logging.TensorboardWriter method) add_hparams() (helios.core.logging.TensorboardWriter method) add_image() (helios.core.logging.TensorboardWriter method) add_images() (helios.core.logging.TensorboardWriter method) add_pr_curve() (helios.core.logging.TensorboardWriter method)

add_scalar() (helios.core.logging.TensorboardWriter method) add_scalars() (helios.core.logging.TensorboardWriter method) add_text() (helios.core.logging.TensorboardWriter method) all_reduce_tensors() (in module helios.core.distributed) append_metadata_to_chkpt_name() (helios.model.model.Model method) append_to_banner() (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)

bgr2ycbcr() (in module helios.data.functional)

C

calculate_mae() (in module helios.metrics.functional) calculate_mae_torch() (in module helios.metrics.functional) calculate_mAP() (in module helios.metrics.functional) calculate_psnr() (in module helios.metrics.functional) calculate_psnr_torch() (in module helios.metrics.functional) calculate_ssim() (in module helios.metrics.functional) calculate_ssim_torch() (in module helios.metrics.functional) CalculateMAE (class in helios.metrics.metrics) CalculateMAP (class in helios.metrics.metrics) CalculatePSNR (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) close() (helios.core.logging.RootLogger method) (helios.core.logging.TensorboardWriter method) close_default_loggers() (in module helios.core.logging) collate_fn (helios.data.datamodule.DataLoaderParams attribute) COLLATE_FN_REGISTRY (in module helios.data.datamodule) configure_model() (helios.plugins.optuna.plugin.OptunaPlugin method) (helios.plugins.plugin.Plugin method) configure_trainer() (helios.plugins.optuna.plugin.OptunaPlugin method) (helios.plugins.plugin.CUDAPlugin 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_loggers() (in module helios.core.logging) create_default_numpy_rng() (in module helios.core.rng) 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_scheduler() (in module helios.scheduler.utils) create_study_starting_from_trial() (in module helios.plugins.optuna.utils) create_transform() (in module helios.data.transforms) CUDAPlugin (class in helios.plugins.plugin) 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) debug_mode (helios.data.datamodule.DataLoaderParams attribute) 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)

E

early_stop_count (helios.trainer.TrainingState attribute) EMA (class in helios.nn.swa_utils) EPOCH (helios.trainer.TrainingUnit attribute)

error() (helios.core.logging.RootLogger method) eval() (helios.model.model.Model method) exception() (helios.core.logging.RootLogger method) export_to_onnx() (in module helios.onnx)

F

find_last_checkpoint() (in module helios.trainer) find_pretrained_file() (in module helios.model.utils) fit() (helios.trainer.Trainer method) flush() (helios.core.logging.RootLogger method) (helios.core.logging.TensorboardWriter method) flush_default_loggers() (in module helios.core.logging) forward() (helios.data.transforms.ToImageTensor method) (helios.losses.weighted_loss.WeightedLoss method) (helios.metrics.metrics.CalculateMAE method) (helios.metrics.metrics.CalculateMAP method) (helios.metrics.metrics.CalculatePSNR 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.logging) 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_lr() (helios.scheduler.schedulers.CosineAnnealingRestartLR method) (helios.scheduler.schedulers.MultiStepRestartLR method) get_rng_state_dict() (in module helios.core.rng) get_root_logger() (in module helios.core.logging) get_tensorboard_writer() (in module helios.core.logging) get_trainer_safe_types_for_load() (in module helios.trainer) 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.logging 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.logging.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.logging) is_using_torchrun() (in module helios.core.distributed) ITERATION (helios.trainer.TrainingUnit attribute)

K

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

L

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.rng.DefaultNumpyRNG method) (helios.model.model.Model method) (helios.plugins.plugin.Plugin 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.logging.RootLogger property) logger (helios.core.logging.RootLogger property) 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.logging 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) MultiStepRestartLR (class in helios.scheduler.schedulers)

N

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) pin_memory (helios.data.datamodule.DataLoaderParams attribute) Plugin (class in helios.plugins.plugin) plugin_id (helios.plugins.optuna.plugin.OptunaPlugin attribute) (helios.plugins.plugin.CUDAPlugin attribute) PLUGIN_REGISTRY (in module helios.plugins.plugin) plugins (helios.trainer.Trainer property)

prepare_data() (helios.data.datamodule.DataModule method) process_testing_batch() (helios.plugins.plugin.CUDAPlugin method) (helios.plugins.plugin.Plugin method) process_training_batch() (helios.plugins.plugin.CUDAPlugin method) (helios.plugins.plugin.Plugin method) process_validation_batch() (helios.plugins.plugin.CUDAPlugin method) (helios.plugins.plugin.Plugin method) 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_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_default_loggers() (in module helios.core.logging) 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) RootLogger (class in helios.core.logging) run_path (helios.core.logging.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) 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.logging.RootLogger method) (helios.core.logging.TensorboardWriter method) (helios.data.datamodule.DataModule method) (helios.model.model.Model method) (helios.plugins.optuna.plugin.OptunaPlugin method) (helios.plugins.plugin.CUDAPlugin method) (helios.plugins.plugin.Plugin method) setup_default_loggers() (in module helios.core.logging) 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.rng.DefaultNumpyRNG method) (helios.model.model.Model method) (helios.plugins.optuna.plugin.OptunaPlugin method) (helios.plugins.plugin.Plugin 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) TensorboardWriter (class in helios.core.logging) TEST (helios.data.datamodule.DatasetSplit 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) 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) 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)

V

VALID (helios.data.datamodule.DatasetSplit 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

warning() (helios.core.logging.RootLogger method)

WeightedLoss (class in helios.losses.weighted_loss) world_size (helios.core.distributed.DistributedInfo attribute)