Index

A

absolute_avg_epoch_time_plot() (in module itwinai.scalability_report.plot) active_run (itwinai.loggers.MLFlowLogger attribute) Adapter (class in itwinai.components) add_epoch_time() (itwinai.loggers.EpochTimeTracker method) 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) barrier() (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)

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

calculate_comp_and_comm_time() (in module itwinai.scalability_report.utils) calculate_gpu_statistics() (in module itwinai.scalability_report.utils) calculate_plot_dimensions() (in module itwinai.scalability_report.plot) callbacks (itwinai.tensorflow.trainer.TensorflowTrainer attribute) check_contains_columns() (in module itwinai.scalability_report.utils) check_initialized() (in module itwinai.loggers) (in module itwinai.torch.distributed) check_not_initialized() (in module itwinai.loggers) check_probing_interval_consistency() (in module itwinai.scalability_report.utils) 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)

communication_data_report() (in module itwinai.scalability_report.reports) computation_fraction_bar_plot() (in module itwinai.scalability_report.plot) compute_metrics() (itwinai.torch.trainer.TorchTrainer method) Configuration (class in itwinai.torch.config) ConsoleLogger (class in itwinai.loggers) create_dataloader() (itwinai.torch.distributed.TorchDistributedStrategy method) create_dataloaders() (itwinai.torch.inference.TorchPredictor method) (itwinai.torch.trainer.TorchTrainer method) create_logger_context() (itwinai.loggers.ConsoleLogger method) (itwinai.loggers.EmptyLogger 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.gan.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.EmptyLogger 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.TorchTrainer property) device() (itwinai.torch.distributed.TorchDistributedStrategy method)

discriminator (itwinai.torch.gan.GANTrainer attribute) dist_backend (itwinai.torch.config.TrainingConfiguration attribute) distribute_model() (itwinai.torch.inference.TorchPredictor 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.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

EmptyLogger (class in itwinai.loggers) epoch_time_report() (in module itwinai.scalability_report.reports) epochs (itwinai.tensorflow.trainer.TensorflowTrainer attribute) EpochTimeTracker (class in itwinai.loggers) 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.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.gan) GANTrainingConfiguration (class in itwinai.torch.gan) 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_random_name() (in module itwinai.utils) generator (itwinai.torch.gan.GANTrainer attribute) get_adaptive_ray_scaling_config() (in module itwinai.distributed) get_computation_fraction_data() (in module itwinai.scalability_report.utils) get_default_distributed_kwargs() (itwinai.torch.trainer.TorchTrainer method) get_root_cause() (in module itwinai.utils)

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) gpu_bar_plot() (in module itwinai.scalability_report.plot) gpu_data_report() (in module itwinai.scalability_report.reports) 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

inference_dataloader (itwinai.torch.inference.TorchPredictor attribute) 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.RayDeepSpeedStrategy method) (itwinai.torch.distributed.RayHorovodStrategy method) (itwinai.torch.distributed.TorchDDPStrategy method) (itwinai.torch.distributed.TorchDistributedStrategy method) init_lightning_mlflow() (in module itwinai.torch.mlflow) 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.loggers.Logger attribute) (itwinai.torch.distributed.TorchDistributedStrategy attribute) is_jsonable() (in module itwinai.serialization) is_main_worker (itwinai.torch.distributed.TorchDistributedStrategy property) itwinai.components module itwinai.distributed module itwinai.loggers module itwinai.parser module itwinai.pipeline module itwinai.scalability_report.data module itwinai.scalability_report.plot module itwinai.scalability_report.reports module itwinai.scalability_report.utils 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.gan module itwinai.torch.inference module itwinai.torch.loggers module itwinai.torch.mlflow module itwinai.torch.reproducibility module itwinai.torch.trainer module itwinai.torch.tuning module itwinai.torch.type module itwinai.type module itwinai.utils module ItwinaiLogger (class in itwinai.torch.loggers)

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.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.EmptyLogger 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.inference.TorchPredictor 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.inference.TorchPredictor attribute) (itwinai.torch.trainer.TorchTrainer attribute) LoggersCollection (class in itwinai.loggers) LogMixin (class in itwinai.loggers) loss (itwinai.torch.config.TrainingConfiguration attribute) (itwinai.torch.gan.GANTrainer attribute) (itwinai.torch.gan.GANTrainingConfiguration attribute) (itwinai.torch.trainer.TorchTrainer attribute) lr_scheduler (itwinai.torch.config.TrainingConfiguration attribute) (itwinai.torch.trainer.TorchTrainer attribute) lr_scheduler_discriminator (itwinai.torch.gan.GANTrainer attribute) (itwinai.torch.gan.GANTrainingConfiguration attribute) lr_scheduler_discriminator_gamma (itwinai.torch.gan.GANTrainingConfiguration attribute) lr_scheduler_discriminator_step_size (itwinai.torch.gan.GANTrainingConfiguration attribute) lr_scheduler_gamma (itwinai.torch.config.TrainingConfiguration attribute) lr_scheduler_generator (itwinai.torch.gan.GANTrainer attribute) (itwinai.torch.gan.GANTrainingConfiguration attribute) lr_scheduler_generator_gamma (itwinai.torch.gan.GANTrainingConfiguration attribute) lr_scheduler_generator_step_size (itwinai.torch.gan.GANTrainingConfiguration attribute) lr_scheduler_step_size (itwinai.torch.config.TrainingConfiguration attribute)

M

macro_batch_size (itwinai.tensorflow.trainer.TensorflowTrainer attribute) make_config_paths_absolute() (in module itwinai.utils) max_params_num (itwinai.utils.SignatureInspector property) measure_communication_overhead (itwinai.torch.trainer.TorchTrainer attribute) measure_epoch_time (itwinai.torch.trainer.TorchTrainer attribute) measure_gpu_data (itwinai.torch.trainer.TorchTrainer attribute) 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) 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) (itwinai.torch.gan.GANTrainingConfiguration attribute) model_from_json() (in module itwinai.tensorflow.utils) model_to_json() (in module itwinai.tensorflow.utils) ModelLoader (class in itwinai.serialization) module itwinai.components itwinai.distributed itwinai.loggers itwinai.parser itwinai.pipeline itwinai.scalability_report.data itwinai.scalability_report.plot itwinai.scalability_report.reports itwinai.scalability_report.utils 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.gan itwinai.torch.inference itwinai.torch.loggers itwinai.torch.mlflow itwinai.torch.reproducibility itwinai.torch.trainer itwinai.torch.tuning 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_betas (itwinai.torch.config.TrainingConfiguration attribute) optim_discriminator_betas (itwinai.torch.gan.GANTrainingConfiguration attribute) optim_discriminator_lr (itwinai.torch.gan.GANTrainingConfiguration attribute) optim_discriminator_momentum (itwinai.torch.gan.GANTrainingConfiguration attribute) optim_discriminator_weight_decay (itwinai.torch.gan.GANTrainingConfiguration attribute) optim_generator_betas (itwinai.torch.gan.GANTrainingConfiguration attribute) optim_generator_lr (itwinai.torch.gan.GANTrainingConfiguration attribute) optim_generator_momentum (itwinai.torch.gan.GANTrainingConfiguration attribute) optim_generator_weight_decay (itwinai.torch.gan.GANTrainingConfiguration attribute)

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) optimizer_discriminator (itwinai.torch.gan.GANTrainer attribute) (itwinai.torch.gan.GANTrainingConfiguration attribute) optimizer_generator (itwinai.torch.gan.GANTrainer attribute) (itwinai.torch.gan.GANTrainingConfiguration attribute)

P

parameters (itwinai.components.BaseComponent attribute) (itwinai.serialization.Serializable attribute) pin_gpu_memory (itwinai.torch.config.TrainingConfiguration attribute) Pipeline (class in itwinai.pipeline)

policy (itwinai.components.Adapter attribute) predict() (itwinai.torch.inference.TorchPredictor method) Predictor (class in itwinai.components) profiler (itwinai.torch.trainer.TorchTrainer attribute) Prov4MLLogger (class in itwinai.loggers)

R

ray_cluster_is_running() (in module itwinai.distributed) ray_report() (itwinai.torch.trainer.TorchTrainer method) RayDDPStrategy (class in itwinai.torch.distributed) RayDeepSpeedStrategy (class in itwinai.torch.distributed) RayHorovodStrategy (class in itwinai.torch.distributed) RayTorchDistributedStrategy (class in itwinai.torch.distributed) read_scalability_metrics_from_csv() (in module itwinai.scalability_report.data) RegressionTorchPredictor (class in itwinai.torch.inference)

relative_epoch_time_speedup_plot() (in module itwinai.scalability_report.plot) 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) (itwinai.torch.trainer.TorchTrainer attribute) run_sanity_check() (in module itwinai.tests.sanity_check)

S

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.gan.GANTrainer method) (itwinai.torch.trainer.TorchTrainer method) save_dir (itwinai.torch.loggers.ItwinaiLogger property) save_fake_generator_images() (itwinai.torch.gan.GANTrainer method) save_hyperparameters() (itwinai.loggers.ConsoleLogger method) (itwinai.loggers.EmptyLogger 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) search_space() (in module itwinai.torch.tuning) 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.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) 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.trainer.TorchTrainer 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_uri() (in module itwinai.utils) to_yaml() (itwinai.serialization.Serializable method) torch_rng (itwinai.torch.inference.TorchPredictor attribute) (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.gan.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.gan.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.gan.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.gan.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)

Z

z_dim (itwinai.torch.gan.GANTrainingConfiguration attribute)