Index

A

active_run (itwinai.loggers.MLFlowLogger attribute) Adapter (class in itwinai.components) add_epoch_time() (itwinai.loggers.EpochTimeTracker method) add_replace_field() (in module itwinai.parser) after_save_checkpoint() (itwinai.torch.loggers.ItwinaiLogger method)

allgather_obj() (itwinai.torch.distributed.DeepSpeedStrategy method) (itwinai.torch.distributed.HorovodStrategy method) (itwinai.torch.distributed.NonDistributedStrategy method) (itwinai.torch.distributed.TorchDDPStrategy method) (itwinai.torch.distributed.TorchDistributedStrategy method) ArgumentParser (class in itwinai.parser)

B

backend (itwinai.torch.distributed.DeepSpeedStrategy attribute) (itwinai.torch.distributed.TorchDDPStrategy attribute) BaseComponent (class in itwinai.components)

Batch (in module itwinai.torch.type) batch_size (itwinai.torch.config.TrainingConfiguration attribute) builtin_print() (in module itwinai.distributed)

C

callbacks (itwinai.tensorflow.trainer.TensorflowTrainer attribute) check_initialized() (in module itwinai.torch.distributed) checkpoint_and_report() (itwinai.torch.trainer.RayTorchTrainer method) clean_up() (itwinai.torch.distributed.DeepSpeedStrategy method) (itwinai.torch.distributed.HorovodStrategy method) (itwinai.torch.distributed.NonDistributedStrategy method) (itwinai.torch.distributed.TorchDDPStrategy method) (itwinai.torch.distributed.TorchDistributedStrategy method) cleanup() (itwinai.components.BaseComponent method) clear_key() (in module itwinai.utils) close_logger() (itwinai.torch.trainer.RayTorchTrainer method) ClusterEnvironment (class in itwinai.distributed) compute_metrics() (itwinai.torch.trainer.TorchTrainer method) config (itwinai.parser.ConfigParser attribute) ConfigParser (class in itwinai.parser)

Configuration (class in itwinai.torch.config) ConsoleLogger (class in itwinai.loggers) create_dataloader() (itwinai.torch.distributed.RayDDPStrategy method) (itwinai.torch.distributed.TorchDistributedStrategy method) create_dataloaders() (itwinai.torch.trainer.RayTorchTrainer method) (itwinai.torch.trainer.TorchTrainer method) create_logger_context() (itwinai.loggers.ConsoleLogger method) (itwinai.loggers.Logger method) (itwinai.loggers.LoggersCollection method) (itwinai.loggers.MLFlowLogger method) (itwinai.loggers.Prov4MLLogger method) (itwinai.loggers.TensorBoardLogger method) (itwinai.loggers.WandBLogger method) create_model_loss_optimizer() (itwinai.torch.trainer.GANTrainer method) (itwinai.torch.trainer.TorchTrainer method)

D

DataGetter (class in itwinai.components) DataProcessor (class in itwinai.components) DataSplitter (class in itwinai.components) DeepSpeedStrategy (class in itwinai.torch.distributed) destroy_logger_context() (itwinai.loggers.ConsoleLogger method) (itwinai.loggers.Logger method) (itwinai.loggers.LoggersCollection method) (itwinai.loggers.MLFlowLogger method) (itwinai.loggers.Prov4MLLogger method) (itwinai.loggers.TensorBoardLogger method) (itwinai.loggers.WandBLogger method) detect_distributed_environment() (in module itwinai.distributed) device (itwinai.torch.trainer.RayTorchTrainer property) (itwinai.torch.trainer.TorchTrainer property)

device() (itwinai.torch.distributed.TorchDistributedStrategy method) dist_backend (itwinai.torch.config.TrainingConfiguration attribute) distributed() (in module itwinai.torch.trainer) (itwinai.torch.distributed.DeepSpeedStrategy method) (itwinai.torch.distributed.HorovodStrategy method) (itwinai.torch.distributed.NonDistributedStrategy method) (itwinai.torch.distributed.RayDDPStrategy method) (itwinai.torch.distributed.TorchDDPStrategy method) (itwinai.torch.distributed.TorchDistributedStrategy method) distributed_patch_print() (in module itwinai.distributed) distributed_resources_available() (in module itwinai.torch.distributed) DistributedStrategy (class in itwinai.distributed) DistributedStrategyError dynamically_import_class() (in module itwinai.utils)

E

epochs (itwinai.tensorflow.trainer.TensorflowTrainer attribute) EpochTimeTracker (class in itwinai.loggers) exec_pipeline() (in module itwinai.cli) execute() (itwinai.components.Adapter method) (itwinai.components.BaseComponent method) (itwinai.components.DataGetter method) (itwinai.components.DataProcessor method) (itwinai.components.DataSplitter method) (itwinai.components.Predictor method) (itwinai.components.Saver method) (itwinai.components.Trainer method) (itwinai.pipeline.Pipeline method) (itwinai.tensorflow.trainer.TensorflowTrainer method) (itwinai.tests.dummy_components.FakeGetter method) (itwinai.tests.dummy_components.FakeGetterExec method) (itwinai.tests.dummy_components.FakePreproc method) (itwinai.tests.dummy_components.FakePreprocExec method) (itwinai.tests.dummy_components.FakeSaver method) (itwinai.tests.dummy_components.FakeSaverExec method) (itwinai.tests.dummy_components.FakeSplitter method) (itwinai.tests.dummy_components.FakeSplitterExec method) (itwinai.tests.dummy_components.FakeTrainer method) (itwinai.tests.dummy_components.FakeTrainerExec method) (itwinai.torch.inference.TorchPredictor method) (itwinai.torch.trainer.RayTorchTrainer method) (itwinai.torch.trainer.TorchLightningTrainer method) (itwinai.torch.trainer.TorchTrainer method)

experiment (itwinai.torch.loggers.ItwinaiLogger property) experiment_id (itwinai.loggers.Logger property)

F

FakeGetter (class in itwinai.tests.dummy_components) FakeGetterExec (class in itwinai.tests.dummy_components) FakePreproc (class in itwinai.tests.dummy_components) FakePreprocExec (class in itwinai.tests.dummy_components) FakeSaver (class in itwinai.tests.dummy_components) FakeSaverExec (class in itwinai.tests.dummy_components) FakeSplitter (class in itwinai.tests.dummy_components)

FakeSplitterExec (class in itwinai.tests.dummy_components) FakeTrainer (class in itwinai.tests.dummy_components) FakeTrainerExec (class in itwinai.tests.dummy_components) finalize() (itwinai.torch.loggers.ItwinaiLogger method) flatten_dict() (in module itwinai.utils) fp16_allreduce (itwinai.torch.config.TrainingConfiguration attribute) fullname() (in module itwinai.serialization)

G

GANTrainer (class in itwinai.torch.trainer) gather() (itwinai.torch.distributed.DeepSpeedStrategy method) (itwinai.torch.distributed.HorovodStrategy method) (itwinai.torch.distributed.NonDistributedStrategy method) (itwinai.torch.distributed.TorchDDPStrategy method) (itwinai.torch.distributed.TorchDistributedStrategy method) gather_obj() (itwinai.torch.distributed.DeepSpeedStrategy method) (itwinai.torch.distributed.HorovodStrategy method) (itwinai.torch.distributed.NonDistributedStrategy method) (itwinai.torch.distributed.TorchDDPStrategy method) (itwinai.torch.distributed.TorchDistributedStrategy method) generate_communication_plot() (in module itwinai.cli) generate_gpu_data_plots() (in module itwinai.cli) generate_scalability_plot() (in module itwinai.cli) get_default_distributed_kwargs() (itwinai.torch.trainer.TorchTrainer method) get_root_cause() (in module itwinai.parser)

get_strategy() (in module itwinai.tensorflow.distributed) global_rank (itwinai.distributed.ClusterEnvironment attribute) global_rank() (itwinai.torch.distributed.DeepSpeedStrategy method) (itwinai.torch.distributed.HorovodStrategy method) (itwinai.torch.distributed.NonDistributedStrategy method) (itwinai.torch.distributed.RayDDPStrategy method) (itwinai.torch.distributed.TorchDDPStrategy method) (itwinai.torch.distributed.TorchDistributedStrategy method) global_world_size (itwinai.distributed.ClusterEnvironment attribute) global_world_size() (itwinai.torch.distributed.DeepSpeedStrategy method) (itwinai.torch.distributed.HorovodStrategy method) (itwinai.torch.distributed.NonDistributedStrategy method) (itwinai.torch.distributed.RayDDPStrategy method) (itwinai.torch.distributed.TorchDDPStrategy method) (itwinai.torch.distributed.TorchDistributedStrategy method) gradient_predivide_factor (itwinai.torch.config.TrainingConfiguration attribute)

H

has_kwargs (itwinai.utils.SignatureInspector property)

has_varargs (itwinai.utils.SignatureInspector property) HorovodStrategy (class in itwinai.torch.distributed)

I

INFTY (itwinai.utils.SignatureInspector attribute) init() (itwinai.torch.distributed.DeepSpeedStrategy method) (itwinai.torch.distributed.HorovodStrategy method) (itwinai.torch.distributed.NonDistributedStrategy method) (itwinai.torch.distributed.RayDDPStrategy method) (itwinai.torch.distributed.TorchDDPStrategy method) (itwinai.torch.distributed.TorchDistributedStrategy method) init_lightning_mlflow() (in module itwinai.torch.mlflow) initialize_logger() (itwinai.torch.trainer.RayTorchTrainer method) initialize_ray() (in module itwinai.torch.distributed) INPUT_PREFIX (itwinai.components.Adapter attribute) instantiate_callbacks() (itwinai.tensorflow.trainer.TensorflowTrainer static method) instantiate_compile_conf() (itwinai.tensorflow.trainer.TensorflowTrainer static method) is_distributed (itwinai.torch.distributed.NonDistributedStrategy attribute) (itwinai.torch.distributed.TorchDistributedStrategy attribute) is_initialized (itwinai.torch.distributed.TorchDistributedStrategy attribute) is_jsonable() (in module itwinai.serialization) is_main_worker (itwinai.torch.distributed.TorchDistributedStrategy property) itwinai.cli module itwinai.components module itwinai.distributed module itwinai.loggers module itwinai.parser module itwinai.pipeline module itwinai.serialization module

itwinai.tensorflow.distributed module itwinai.tensorflow.trainer module itwinai.tensorflow.utils module itwinai.tests.dummy_components module itwinai.tests.exceptions module itwinai.tests.sanity_check module itwinai.torch.config module itwinai.torch.distributed module itwinai.torch.inference module itwinai.torch.loggers module itwinai.torch.mlflow module itwinai.torch.reproducibility module itwinai.torch.trainer module itwinai.torch.type module itwinai.type module itwinai.utils module ItwinaiLogger (class in itwinai.torch.loggers)

K

kill_mlflow_server() (in module itwinai.cli)

L

load_checkpoint() (itwinai.torch.trainer.GANTrainer method) (itwinai.torch.trainer.TorchTrainer method) load_yaml() (in module itwinai.utils) local_rank (itwinai.distributed.ClusterEnvironment attribute) local_rank() (itwinai.torch.distributed.DeepSpeedStrategy method) (itwinai.torch.distributed.HorovodStrategy method) (itwinai.torch.distributed.NonDistributedStrategy method) (itwinai.torch.distributed.RayDDPStrategy method) (itwinai.torch.distributed.TorchDDPStrategy method) (itwinai.torch.distributed.TorchDistributedStrategy method) local_world_size (itwinai.distributed.ClusterEnvironment attribute) local_world_size() (itwinai.torch.distributed.DeepSpeedStrategy method) (itwinai.torch.distributed.HorovodStrategy method) (itwinai.torch.distributed.NonDistributedStrategy method) (itwinai.torch.distributed.RayDDPStrategy method) (itwinai.torch.distributed.TorchDDPStrategy method) (itwinai.torch.distributed.TorchDistributedStrategy method) locals2params() (itwinai.serialization.Serializable static method) log() (itwinai.loggers.ConsoleLogger method) (itwinai.loggers.LoggersCollection method) (itwinai.loggers.LogMixin method) (itwinai.loggers.MLFlowLogger method) (itwinai.loggers.Prov4MLLogger method) (itwinai.loggers.TensorBoardLogger method) (itwinai.loggers.WandBLogger method) (itwinai.torch.trainer.RayTorchTrainer method) (itwinai.torch.trainer.TorchTrainer method)

log_freq (itwinai.loggers.Logger property) log_hyperparams() (itwinai.torch.loggers.ItwinaiLogger method) log_metrics() (itwinai.torch.loggers.ItwinaiLogger method) Logger (class in itwinai.loggers) logger (itwinai.torch.trainer.TorchTrainer attribute) LoggersCollection (class in itwinai.loggers) LogMixin (class in itwinai.loggers) loss (itwinai.torch.config.TrainingConfiguration attribute) (itwinai.torch.trainer.TorchTrainer attribute) lr_scheduler (itwinai.torch.trainer.TorchTrainer attribute) LrScheduler (in module itwinai.torch.type)

M

macro_batch_size (itwinai.tensorflow.trainer.TensorflowTrainer attribute) max_params_num (itwinai.utils.SignatureInspector property) Metric (in module itwinai.torch.type) metrics (itwinai.torch.trainer.TorchTrainer attribute) micro_batch_size (itwinai.tensorflow.trainer.TensorflowTrainer attribute) min_params_num (itwinai.utils.SignatureInspector property) MLArtifact (class in itwinai.type) MLDataset (class in itwinai.type) mlflow_server() (in module itwinai.cli) mlflow_ui() (in module itwinai.cli) MLFlowLogger (class in itwinai.loggers) MLModel (class in itwinai.type) model (itwinai.components.Predictor attribute) (itwinai.tests.dummy_components.FakeTrainerExec attribute) (itwinai.torch.inference.TorchPredictor attribute) (itwinai.torch.trainer.TorchTrainer attribute) model_config (itwinai.distributed.ClusterEnvironment attribute) (itwinai.torch.config.Configuration attribute) (itwinai.torch.config.TrainingConfiguration attribute) model_from_json() (in module itwinai.tensorflow.utils) model_to_json() (in module itwinai.tensorflow.utils) ModelLoader (class in itwinai.serialization) module itwinai.cli itwinai.components itwinai.distributed itwinai.loggers itwinai.parser itwinai.pipeline itwinai.serialization itwinai.tensorflow.distributed itwinai.tensorflow.trainer itwinai.tensorflow.utils itwinai.tests.dummy_components itwinai.tests.exceptions itwinai.tests.sanity_check itwinai.torch.config itwinai.torch.distributed itwinai.torch.inference itwinai.torch.loggers itwinai.torch.mlflow itwinai.torch.reproducibility itwinai.torch.trainer itwinai.torch.type itwinai.type itwinai.utils

monitor_exec() (in module itwinai.components) MulticlassTorchPredictor (class in itwinai.torch.inference) MultilabelTorchPredictor (class in itwinai.torch.inference)

N

name (itwinai.components.BaseComponent property) (itwinai.torch.distributed.TorchDistributedStrategy attribute) (itwinai.torch.loggers.ItwinaiLogger property)

NonDistributedStrategy (class in itwinai.torch.distributed) num_workers (itwinai.tensorflow.trainer.TensorflowTrainer attribute) num_workers_dataloader (itwinai.torch.config.TrainingConfiguration attribute)

O

optim_lr (itwinai.torch.config.TrainingConfiguration attribute) optim_momentum (itwinai.torch.config.TrainingConfiguration attribute)

optim_weight_decay (itwinai.torch.config.TrainingConfiguration attribute) optimizer (itwinai.torch.config.TrainingConfiguration attribute) (itwinai.torch.trainer.TorchTrainer attribute)

P

parameters (itwinai.components.BaseComponent attribute) (itwinai.serialization.Serializable attribute) parse_pipeline() (itwinai.parser.ConfigParser method) parse_step() (itwinai.parser.ConfigParser method) pin_gpu_memory (itwinai.torch.config.TrainingConfiguration attribute)

Pipeline (class in itwinai.pipeline) pipeline (itwinai.parser.ConfigParser attribute) policy (itwinai.components.Adapter attribute) Predictor (class in itwinai.components) profiler (itwinai.torch.trainer.TorchTrainer attribute) Prov4MLLogger (class in itwinai.loggers)

R

RayDDPStrategy (class in itwinai.torch.distributed) RayDeepSpeedStrategy (class in itwinai.torch.distributed) RayTorchTrainer (class in itwinai.torch.trainer) RegressionTorchPredictor (class in itwinai.torch.inference) required_params (itwinai.utils.SignatureInspector property)

result (itwinai.tests.dummy_components.FakeGetterExec attribute) (itwinai.tests.dummy_components.FakeSplitterExec attribute) rnd_seed (itwinai.tensorflow.trainer.TensorflowTrainer attribute) run_id (itwinai.loggers.Logger property) run_sanity_check() (in module itwinai.tests.sanity_check)

S

sanity_check() (in module itwinai.cli) sanity_check_all() (in module itwinai.tests.sanity_check) sanity_check_slim() (in module itwinai.tests.sanity_check) sanity_check_tensorflow() (in module itwinai.tests.sanity_check) sanity_check_torch() (in module itwinai.tests.sanity_check) SanityCheckError save() (itwinai.loggers.EpochTimeTracker method) save_checkpoint() (itwinai.torch.trainer.GANTrainer method) (itwinai.torch.trainer.TorchTrainer method) save_dir (itwinai.torch.loggers.ItwinaiLogger property) save_fake_generator_images() (itwinai.torch.trainer.GANTrainer method) save_hyperparameters() (itwinai.loggers.ConsoleLogger method) (itwinai.loggers.Logger method) (itwinai.loggers.LoggersCollection method) (itwinai.loggers.MLFlowLogger method) (itwinai.loggers.Prov4MLLogger method) (itwinai.loggers.TensorBoardLogger method) (itwinai.loggers.WandBLogger method) save_parameters() (itwinai.serialization.Serializable method) savedir (itwinai.loggers.Logger attribute) Saver (class in itwinai.components) seed_worker() (in module itwinai.torch.reproducibility) Serializable (class in itwinai.serialization) SerializationError serialize() (itwinai.loggers.Logger method)

set_device() (itwinai.torch.distributed.TorchDistributedStrategy method) set_epoch() (itwinai.torch.trainer.RayTorchTrainer method) (itwinai.torch.trainer.TorchTrainer method) set_seed() (in module itwinai.torch.reproducibility) should_log() (itwinai.loggers.Logger method) (itwinai.loggers.LoggersCollection method) shuffle_buffer (itwinai.tensorflow.trainer.TensorflowTrainer attribute) shuffle_test (itwinai.torch.config.TrainingConfiguration attribute) shuffle_train (itwinai.torch.config.TrainingConfiguration attribute) shuffle_validation (itwinai.torch.config.TrainingConfiguration attribute) SignatureInspector (class in itwinai.utils) start_logging() (itwinai.loggers.Logger method) steps (itwinai.pipeline.Pipeline attribute) str_to_slice() (in module itwinai.utils) strategy (itwinai.tensorflow.trainer.TensorflowTrainer attribute) (itwinai.torch.trainer.TorchTrainer property) supported_kinds (itwinai.loggers.ConsoleLogger attribute) (itwinai.loggers.Logger attribute) (itwinai.loggers.LoggersCollection attribute) (itwinai.loggers.MLFlowLogger attribute) (itwinai.loggers.Prov4MLLogger attribute) (itwinai.loggers.TensorBoardLogger attribute) (itwinai.loggers.WandBLogger attribute) suppress_workers_output() (in module itwinai.distributed) suppress_workers_print() (in module itwinai.distributed)

T

teardown_lightning_mlflow() (in module itwinai.torch.mlflow) TensorBoardLogger (class in itwinai.loggers) TensorflowTrainer (class in itwinai.tensorflow.trainer) test_dataloader (itwinai.torch.inference.TorchPredictor attribute) (itwinai.torch.trainer.TorchTrainer attribute) test_dataset (itwinai.torch.inference.TorchPredictor attribute) test_epoch() (itwinai.torch.trainer.TorchTrainer method) test_glob_step (itwinai.torch.trainer.TorchTrainer attribute) test_proportion (itwinai.components.DataSplitter property) test_step() (itwinai.torch.trainer.TorchTrainer method) threshold (itwinai.torch.inference.MultilabelTorchPredictor attribute) to_dict() (itwinai.serialization.Serializable method) to_json() (itwinai.serialization.Serializable method) to_yaml() (itwinai.serialization.Serializable method) torch_rng (itwinai.torch.trainer.TorchTrainer attribute) TorchDDPStrategy (class in itwinai.torch.distributed) TorchDistributedStrategy (class in itwinai.torch.distributed)

TorchLightningTrainer (class in itwinai.torch.trainer) TorchModelLoader (class in itwinai.torch.inference) TorchPredictor (class in itwinai.torch.inference) TorchTrainer (class in itwinai.torch.trainer) train() (itwinai.torch.trainer.TorchTrainer method) train_dataloader (itwinai.torch.trainer.TorchTrainer attribute) train_epoch() (itwinai.torch.trainer.GANTrainer method) (itwinai.torch.trainer.TorchTrainer method) train_glob_step (itwinai.torch.trainer.TorchTrainer attribute) train_proportion (itwinai.components.DataSplitter property) train_step() (itwinai.torch.trainer.GANTrainer method) (itwinai.torch.trainer.TorchTrainer method) Trainer (class in itwinai.components) TrainingConfiguration (class in itwinai.torch.config) transform_predictions() (itwinai.torch.inference.MulticlassTorchPredictor method) (itwinai.torch.inference.MultilabelTorchPredictor method) (itwinai.torch.inference.RegressionTorchPredictor method) (itwinai.torch.inference.TorchPredictor method)

U

UninitializedStrategyError

update_parameters() (itwinai.serialization.Serializable method) use_adasum (itwinai.torch.config.TrainingConfiguration attribute)

V

validate_args() (itwinai.pipeline.Pipeline static method) validation_dataloader (itwinai.torch.trainer.TorchTrainer attribute) validation_epoch() (itwinai.torch.trainer.GANTrainer method) (itwinai.torch.trainer.TorchTrainer method)

validation_glob_step (itwinai.torch.trainer.TorchTrainer attribute) validation_proportion (itwinai.components.DataSplitter property) validation_step() (itwinai.torch.trainer.GANTrainer method) (itwinai.torch.trainer.TorchTrainer method) version (itwinai.torch.loggers.ItwinaiLogger property)

W

WandBLogger (class in itwinai.loggers)

worker_rank (itwinai.loggers.Logger attribute)