itwinai
- itwinai.cli
- itwinai.components
- itwinai.distributed
- itwinai.loggers
- itwinai.parser
- itwinai.pipeline
- itwinai.serialization
- itwinai.tests
- itwinai.tensorflow
- distributed.py
- 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
- itwinai.torch
- config.py
ConfigurationTrainingConfigurationTrainingConfiguration.batch_sizeTrainingConfiguration.shuffle_trainTrainingConfiguration.shuffle_validationTrainingConfiguration.shuffle_testTrainingConfiguration.pin_gpu_memoryTrainingConfiguration.num_workers_dataloaderTrainingConfiguration.lossTrainingConfiguration.optimizerTrainingConfiguration.optim_lrTrainingConfiguration.optim_momentumTrainingConfiguration.optim_weight_decayTrainingConfiguration.fp16_allreduceTrainingConfiguration.use_adasumTrainingConfiguration.gradient_predivide_factorTrainingConfiguration.dist_backendTrainingConfiguration.model_config
- distributed.py
distributed_resources_available()check_initialized()initialize_ray()TorchDistributedStrategyTorchDistributedStrategy.is_distributedTorchDistributedStrategy.is_initializedTorchDistributedStrategy.nameTorchDistributedStrategy.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()TorchDistributedStrategy.gather()
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()TorchDDPStrategy.gather()
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()DeepSpeedStrategy.gather()
HorovodStrategyNonDistributedStrategyNonDistributedStrategy.is_distributedNonDistributedStrategy.init()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()NonDistributedStrategy.gather()
RayDDPStrategyRayDeepSpeedStrategy
- inference.py
- loggers.py
- mlflow.py
- reproducibility.py
- type.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.profilerTorchTrainer.strategyTorchTrainer.deviceTorchTrainer.get_default_distributed_kwargs()TorchTrainer.create_model_loss_optimizer()TorchTrainer.create_dataloaders()TorchTrainer.execute()TorchTrainer.set_epoch()TorchTrainer.log()TorchTrainer.save_checkpoint()TorchTrainer.load_checkpoint()TorchTrainer.compute_metrics()TorchTrainer.train()TorchTrainer.train_epoch()TorchTrainer.train_step()TorchTrainer.validation_epoch()TorchTrainer.validation_step()TorchTrainer.test_epoch()TorchTrainer.test_step()
GANTrainerTorchLightningTrainerdistributed()RayTorchTrainer
- config.py
- itwinai.type
- itwinai.utils