itwinai
- itwinai PyTorch Modules
- distributed.py
distributed_resources_available()TorchDistributedStrategyTorchDistributedStrategy.is_distributedTorchDistributedStrategy.is_initializedTorchDistributedStrategy.is_main_workerTorchDistributedStrategy.init()TorchDistributedStrategy.distributed()TorchDistributedStrategy.global_world_size()TorchDistributedStrategy.local_world_size()TorchDistributedStrategy.global_rank()TorchDistributedStrategy.local_rank()TorchDistributedStrategy.device()TorchDistributedStrategy.set_device()TorchDistributedStrategy.create_dataloader()TorchDistributedStrategy.clean_up()TorchDistributedStrategy.allgather_obj()TorchDistributedStrategy.gather_obj()
TorchDDPStrategyTorchDDPStrategy.backendTorchDDPStrategy.init()TorchDDPStrategy.distributed()TorchDDPStrategy.global_world_size()TorchDDPStrategy.local_world_size()TorchDDPStrategy.global_rank()TorchDDPStrategy.local_rank()TorchDDPStrategy.clean_up()TorchDDPStrategy.allgather_obj()TorchDDPStrategy.gather_obj()
DeepSpeedStrategyDeepSpeedStrategy.backendDeepSpeedStrategy.init()DeepSpeedStrategy.distributed()DeepSpeedStrategy.global_world_size()DeepSpeedStrategy.local_world_size()DeepSpeedStrategy.global_rank()DeepSpeedStrategy.local_rank()DeepSpeedStrategy.clean_up()DeepSpeedStrategy.allgather_obj()DeepSpeedStrategy.gather_obj()
HorovodStrategyNonDistributedStrategyNonDistributedStrategy.is_distributedNonDistributedStrategy.init()NonDistributedStrategy.device()NonDistributedStrategy.distributed()NonDistributedStrategy.global_world_size()NonDistributedStrategy.local_world_size()NonDistributedStrategy.global_rank()NonDistributedStrategy.local_rank()NonDistributedStrategy.clean_up()NonDistributedStrategy.allgather_obj()NonDistributedStrategy.gather_obj()
- inference.py
- mlflow.py
- trainer.py
TorchTrainerTorchTrainer.train_dataloaderTorchTrainer.validation_dataloaderTorchTrainer.test_dataloaderTorchTrainer.lossTorchTrainer.optimizerTorchTrainer.lr_schedulerTorchTrainer.torch_rngTorchTrainer.train_glob_stepTorchTrainer.validation_glob_stepTorchTrainer.test_glob_stepTorchTrainer.modelTorchTrainer.loggerTorchTrainer.metricsTorchTrainer.strategyTorchTrainer.deviceTorchTrainer.create_model_loss_optimizer()TorchTrainer.create_dataloaders()TorchTrainer.execute()TorchTrainer.set_epoch()TorchTrainer.log()TorchTrainer.save_checkpoint()TorchTrainer.load_checkpoint()TorchTrainer.train()TorchTrainer.compute_metrics()TorchTrainer.training_step()TorchTrainer.validation_step()TorchTrainer.train_epoch()TorchTrainer.validation_epoch()TorchTrainer.test_epoch()
TorchLightningTrainerpreproc_dataloader()distributed()
- type.py
- reproducibility.py
- config.py
ConfigurationTrainingConfigurationTrainingConfiguration.batch_sizeTrainingConfiguration.shuffle_trainTrainingConfiguration.shuffle_validationTrainingConfiguration.shuffle_testTrainingConfiguration.pin_memoryTrainingConfiguration.num_workersTrainingConfiguration.lrTrainingConfiguration.momentumTrainingConfiguration.fp16_allreduceTrainingConfiguration.use_adasumTrainingConfiguration.gradient_predivide_factorTrainingConfiguration.model_computed_fieldsTrainingConfiguration.model_configTrainingConfiguration.model_fields
- distributed.py
- itwinai Tensorflow Modules
- trainer.py
TensorflowTrainerTensorflowTrainer.epochsTensorflowTrainer.micro_batch_sizeTensorflowTrainer.shuffle_bufferTensorflowTrainer.rnd_seedTensorflowTrainer.callbacksTensorflowTrainer.strategyTensorflowTrainer.num_workersTensorflowTrainer.macro_batch_sizeTensorflowTrainer.instantiate_compile_conf()TensorflowTrainer.instantiate_callbacks()TensorflowTrainer.execute()
- utils.py
- distributed.py
- trainer.py
- itwinai.cli
- itwinai.components
- itwinai.distributed
- itwinai.loggers
- itwinai.parser
- itwinai.pipeline
- itwinai.serialization
- itwinai.type
- itwinai.utils