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) 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) 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) 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.TorchDistributedStrategy method) create_dataloaders() (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.TensorBoardLogger method) (itwinai.loggers.WanDBLogger method) create_model_loss_optimizer() (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.TensorBoardLogger method) (itwinai.loggers.WanDBLogger method) detect_distributed_environment() (in module itwinai.distributed) device (itwinai.torch.trainer.TorchTrainer property)

device() (itwinai.torch.distributed.NonDistributedStrategy method) (itwinai.torch.distributed.TorchDistributedStrategy method) distributed() (in module itwinai.torch.trainer) (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) 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.torch.inference.TorchPredictor method) (itwinai.torch.trainer.TorchLightningTrainer method) (itwinai.torch.trainer.TorchTrainer method)

F

flatten_dict() (in module itwinai.utils)

fp16_allreduce (itwinai.torch.config.TrainingConfiguration attribute) fullname() (in module itwinai.serialization)

G

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) 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.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.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.TorchDDPStrategy method) (itwinai.torch.distributed.TorchDistributedStrategy method) init_lightning_mlflow() (in module itwinai.torch.mlflow) 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.torch.config module itwinai.torch.distributed module itwinai.torch.inference module itwinai.torch.mlflow module itwinai.torch.reproducibility module itwinai.torch.trainer module itwinai.torch.type module itwinai.type module itwinai.utils module

K

kill_mlflow_server() (in module itwinai.cli)

L

load_checkpoint() (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.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.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.TensorBoardLogger method) (itwinai.loggers.WanDBLogger method) (itwinai.torch.trainer.TorchTrainer method)

log_freq (itwinai.loggers.Logger property) Logger (class in itwinai.loggers) logger (itwinai.torch.trainer.TorchTrainer attribute) LoggersCollection (class in itwinai.loggers) LogMixin (class in itwinai.loggers) Loss (in module itwinai.torch.type) loss (itwinai.torch.trainer.TorchTrainer attribute) lr (itwinai.torch.config.TrainingConfiguration 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.torch.inference.TorchPredictor attribute) (itwinai.torch.trainer.TorchTrainer attribute) model_computed_fields (itwinai.distributed.ClusterEnvironment attribute) (itwinai.torch.config.Configuration attribute) (itwinai.torch.config.TrainingConfiguration attribute) model_config (itwinai.distributed.ClusterEnvironment attribute) (itwinai.torch.config.Configuration attribute) (itwinai.torch.config.TrainingConfiguration attribute) model_fields (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.torch.config itwinai.torch.distributed itwinai.torch.inference itwinai.torch.mlflow itwinai.torch.reproducibility itwinai.torch.trainer itwinai.torch.type itwinai.type itwinai.utils momentum (itwinai.torch.config.TrainingConfiguration attribute) monitor_exec() (in module itwinai.components) MulticlassTorchPredictor (class in itwinai.torch.inference) MultilabelTorchPredictor (class in itwinai.torch.inference)

N

name (itwinai.components.BaseComponent property) NonDistributedStrategy (class in itwinai.torch.distributed)

num_workers (itwinai.tensorflow.trainer.TensorflowTrainer attribute) (itwinai.torch.config.TrainingConfiguration attribute)

O

optimizer (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_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) preproc_dataloader() (in module itwinai.torch.trainer)

R

RegressionTorchPredictor (class in itwinai.torch.inference)

required_params (itwinai.utils.SignatureInspector property) rnd_seed (itwinai.tensorflow.trainer.TensorflowTrainer attribute)

S

save() (itwinai.loggers.EpochTimeTracker method) save_checkpoint() (itwinai.torch.trainer.TorchTrainer method) save_hyperparameters() (itwinai.loggers.ConsoleLogger method) (itwinai.loggers.Logger method) (itwinai.loggers.LoggersCollection method) (itwinai.loggers.MLFlowLogger 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) scalability_report() (in module itwinai.cli) 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_seed() (in module itwinai.torch.reproducibility) should_log() (itwinai.loggers.Logger method) (itwinai.loggers.LoggersCollection method) shuffle_buffer (itwinai.tensorflow.trainer.TensorflowTrainer 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_types (itwinai.loggers.Logger attribute)

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) 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.TorchTrainer method) train_glob_step (itwinai.torch.trainer.TorchTrainer attribute) train_proportion (itwinai.components.DataSplitter property) Trainer (class in itwinai.components) training_step() (itwinai.torch.trainer.TorchTrainer method) 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.TorchTrainer method)

validation_glob_step (itwinai.torch.trainer.TorchTrainer attribute) validation_proportion (itwinai.components.DataSplitter property) validation_step() (itwinai.torch.trainer.TorchTrainer method)

W

WanDBLogger (class in itwinai.loggers)