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

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.OptunaPlugin method) checkpoint_sampler() (in module helios.plugins.optuna) 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) configure_model() (helios.plugins.optuna.OptunaPlugin method) (helios.plugins.plugin.Plugin method)

configure_trainer() (helios.plugins.optuna.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_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_plugin() (in module helios.plugins.plugin) create_sampler() (in module helios.data.samplers) create_scheduler() (in module helios.scheduler.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) 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.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_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) 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.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.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_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.swa_utils helios.nn.utils helios.onnx helios.optim helios.optim.utils helios.plugins helios.plugins.optuna 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.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)

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.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)

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.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) 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) resume_study() (in module helios.plugins.optuna) 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) 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.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.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.OptunaPlugin method) (helios.plugins.plugin.Plugin method) suggest() (helios.plugins.optuna.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.OptunaPlugin property)

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)