Index A | B | C | D | E | F | G | I | J | L | M | N | O | P | R | S | T | U | V | W A ACTIVE (composer.profiler.ProfilerAction attribute) AdamHparams (class in composer.optim.optimizer_hparams) adamw() (composer.optim.decoupled_weight_decay.DecoupledAdamW static method) AdamWHparams (class in composer.optim.optimizer_hparams) add_vision_dataset_transform() (in module composer.datasets.utils) ADE20k (class in composer.datasets.ade20k) ADE20kDatasetHparams (class in composer.datasets.ade20k) AFTER_BACKWARD (composer.core.event.Event attribute) after_backward() (composer.core.callback.Callback method) AFTER_DATALOADER (composer.core.event.Event attribute) after_dataloader() (composer.core.callback.Callback method) AFTER_FORWARD (composer.core.event.Event attribute) after_forward() (composer.core.callback.Callback method) AFTER_LOSS (composer.core.event.Event attribute) after_loss() (composer.core.callback.Callback method) AFTER_TRAIN_BATCH (composer.core.event.Event attribute) after_train_batch() (composer.core.callback.Callback method) Algorithm (class in composer.core.algorithm) AlgorithmHparams (class in composer.algorithms.algorithm_hparams) Alibi (class in composer.algorithms.alibi.alibi) AlibiHparams (class in composer.algorithms.hparams) all_gather() (in module composer.utils.dist) all_gather_object() (in module composer.utils.dist) all_reduce() (in module composer.utils.dist) AMP (composer.core.precision.Precision attribute) apply() (composer.algorithms.blurpool.blurpool.BlurPool method) (composer.algorithms.channels_last.channels_last.ChannelsLast method) (composer.algorithms.cutmix.cutmix.CutMix method) (composer.algorithms.cutout.cutout.CutOut method) (composer.algorithms.factorize.factorize.Factorize method) (composer.algorithms.ghost_batchnorm.ghost_batchnorm.GhostBatchNorm method) (composer.algorithms.layer_freezing.layer_freezing.LayerFreezing method) (composer.algorithms.mixup.mixup.MixUp method) (composer.algorithms.progressive_resizing.progressive_resizing.ProgressiveResizing method) (composer.algorithms.sam.sam.SAM method) (composer.algorithms.scale_schedule.scale_schedule.ScaleSchedule method) (composer.algorithms.selective_backprop.selective_backprop.SelectiveBackprop method) (composer.algorithms.squeeze_excite.squeeze_excite.SqueezeExcite method) (composer.algorithms.stochastic_depth.stochastic_depth.StochasticDepth method) (composer.core.algorithm.Algorithm method) apply_alibi() (in module composer.algorithms.alibi.alibi) apply_blurpool() (in module composer.algorithms.blurpool.blurpool) apply_channels_last() (in module composer.algorithms.channels_last.channels_last) apply_factorization() (in module composer.algorithms.factorize.factorize) apply_ghost_batchnorm() (in module composer.algorithms.ghost_batchnorm.ghost_batchnorm) apply_squeeze_excite() (in module composer.algorithms.squeeze_excite.squeeze_excite) apply_stochastic_depth() (in module composer.algorithms.stochastic_depth.stochastic_depth) as_batch_dict() (in module composer.core.types) as_batch_pair() (in module composer.core.types) AugmentAndMixTransform (class in composer.algorithms.augmix.augmix) augmentation_sets (in module composer.algorithms.utils.augmentation_primitives) AugmentationFn (in module composer.algorithms.utils.augmentation_primitives) AugMix (class in composer.algorithms.augmix.augmix) augmix_image() (in module composer.algorithms.augmix.augmix) AugMixHparams (class in composer.algorithms.hparams) autocontrast() (in module composer.algorithms.utils.augmentation_primitives) B backends (composer.core.logging.logger.Logger attribute) backwards_create_graph (composer.core.algorithm.Algorithm property) barrier() (in module composer.utils.dist) BATCH (composer.core.logging.logger.LogLevel attribute) batch (composer.core.state.State attribute) (composer.core.time.Timer property) BATCH (composer.core.time.TimeUnit attribute) Batch (in module composer.core.types) batch_dict (composer.core.state.State property) BATCH_END (composer.core.event.Event attribute) batch_end() (composer.core.callback.Callback method) batch_idx (composer.core.state.State property) batch_in_epoch (composer.core.time.Timer property) batch_num_samples (composer.core.state.State attribute) batch_num_tokens (composer.core.state.State attribute) batch_pair (composer.core.state.State property) batch_size (composer.core.types.DataLoader attribute) BATCH_START (composer.core.event.Event attribute) batch_start() (composer.core.callback.Callback method) batch_to_device() (composer.trainer.devices.device.Device method) BatchDict (in module composer.core.types) BatchPair (in module composer.core.types) BEFORE_BACKWARD (composer.core.event.Event attribute) before_backward() (composer.core.callback.Callback method) BEFORE_FORWARD (composer.core.event.Event attribute) before_forward() (composer.core.callback.Callback method) BEFORE_LOSS (composer.core.event.Event attribute) before_loss() (composer.core.callback.Callback method) BEFORE_TRAIN_BATCH (composer.core.event.Event attribute) before_train_batch() (composer.core.callback.Callback method) BERTForClassificationHparams (class in composer.models.bert.bert_hparams) BERTHparams (class in composer.models.bert.bert_hparams) BERTModel (class in composer.models.bert.model) BinaryF1Score (class in composer.models.nlp_metrics) blur_2d() (in module composer.algorithms.blurpool.blurpool_layers) BlurConv2d (class in composer.algorithms.blurpool.blurpool_layers) blurmax_pool2d() (in module composer.algorithms.blurpool.blurpool_layers) BlurMaxPool2d (class in composer.algorithms.blurpool.blurpool_layers) BlurPool (class in composer.algorithms.blurpool.blurpool) BlurPool2d (class in composer.algorithms.blurpool.blurpool_layers) BlurPoolHparams (class in composer.algorithms.hparams) BratsDatasetHparams (class in composer.datasets.brats) BreakEpochException brightness() (in module composer.algorithms.utils.augmentation_primitives) brightness_original() (in module composer.algorithms.utils.augmentation_primitives) broadcast() (in module composer.utils.dist) broadcast_object_list() (in module composer.utils.dist) C calculate_same_padding() (in module composer.models.efficientnets) Callback (class in composer.core.callback) CallbackHparams (class in composer.callbacks.callback_hparams) canonical_name (composer.core.event.Event property) ChannelsLast (class in composer.algorithms.channels_last.channels_last) ChannelsLastHparams (class in composer.algorithms.hparams) check_for_index_targets() (in module composer.models.loss) checkpoint_folder (composer.trainer.trainer.Trainer property) CIFAR10_ResNet20 (class in composer.models.resnet20_cifar10.model) CIFAR10_ResNet56 (class in composer.models.resnet56_cifar10.model) CIFAR10_ResNet9 (class in composer.models.resnet9_cifar10.model) CIFAR10DatasetHparams (class in composer.datasets.cifar10) CIFAR_ResNet (class in composer.models.resnets) CIFAR_ResNet.Block (class in composer.models.resnets) CIFARResNet20Hparams (class in composer.models.resnet20_cifar10.resnet20_cifar10_hparams) CIFARResNet9Hparams (class in composer.models.resnet9_cifar10.resnet9_cifar10_hparams) CIFARResNetHparams (class in composer.models.resnet56_cifar10.resnet56_cifar10_hparams) close() (composer.core.callback.Callback method) (composer.core.engine.Engine method) color() (in module composer.algorithms.utils.augmentation_primitives) color_original() (in module composer.algorithms.utils.augmentation_primitives) ColOut (class in composer.algorithms.colout.colout) colout_batch() (in module composer.algorithms.colout.colout) ColOutHparams (class in composer.algorithms.hparams) ColOutTransform (class in composer.algorithms.colout.colout) composer.algorithms module composer.algorithms.algorithm_hparams module composer.algorithms.algorithm_registry module composer.algorithms.alibi module composer.algorithms.alibi.alibi module composer.algorithms.augmix module composer.algorithms.augmix.augmix module composer.algorithms.blurpool module composer.algorithms.blurpool.blurpool module composer.algorithms.blurpool.blurpool_layers module composer.algorithms.channels_last module composer.algorithms.channels_last.channels_last module composer.algorithms.colout module composer.algorithms.colout.colout module composer.algorithms.cutmix module composer.algorithms.cutmix.cutmix module composer.algorithms.cutout module composer.algorithms.cutout.cutout module composer.algorithms.factorize module composer.algorithms.factorize.factorize module composer.algorithms.factorize.factorize_core module composer.algorithms.factorize.factorize_modules module composer.algorithms.ghost_batchnorm module composer.algorithms.ghost_batchnorm.ghost_batchnorm module composer.algorithms.hparams module composer.algorithms.label_smoothing module composer.algorithms.label_smoothing.label_smoothing module composer.algorithms.layer_freezing module composer.algorithms.layer_freezing.layer_freezing module composer.algorithms.mixup module composer.algorithms.mixup.mixup module composer.algorithms.no_op_model module composer.algorithms.no_op_model.no_op_model module composer.algorithms.progressive_resizing module composer.algorithms.progressive_resizing.progressive_resizing module composer.algorithms.randaugment module composer.algorithms.randaugment.randaugment module composer.algorithms.sam module composer.algorithms.sam.sam module composer.algorithms.scale_schedule module composer.algorithms.scale_schedule.scale_schedule module composer.algorithms.selective_backprop module composer.algorithms.selective_backprop.selective_backprop module composer.algorithms.seq_length_warmup module composer.algorithms.seq_length_warmup.seq_length_warmup module composer.algorithms.squeeze_excite module composer.algorithms.squeeze_excite.squeeze_excite module composer.algorithms.stochastic_depth module composer.algorithms.stochastic_depth.sample_stochastic_layers module composer.algorithms.stochastic_depth.stochastic_depth module composer.algorithms.stochastic_depth.stochastic_layers module composer.algorithms.swa module composer.algorithms.swa.swa module composer.algorithms.utils module composer.algorithms.utils.augmentation_common module composer.algorithms.utils.augmentation_primitives module composer.callbacks module composer.callbacks.callback_hparams module composer.callbacks.grad_monitor module composer.callbacks.lr_monitor module composer.callbacks.memory_monitor module composer.callbacks.run_directory_uploader module composer.callbacks.speed_monitor module composer.core module composer.core.algorithm module composer.core.callback module composer.core.data_spec module composer.core.engine module composer.core.evaluator module composer.core.event module composer.core.logging module composer.core.logging.base_backend module composer.core.logging.logger module composer.core.precision module composer.core.serializable module composer.core.state module composer.core.time module composer.core.types module composer.datasets module composer.datasets.ade20k module composer.datasets.brats module composer.datasets.cifar10 module composer.datasets.dataloader module composer.datasets.dataset_registry module composer.datasets.evaluator module composer.datasets.glue module composer.datasets.hparams module composer.datasets.imagenet module composer.datasets.lm_datasets module composer.datasets.mnist module composer.datasets.synthetic module composer.datasets.utils module composer.functional module composer.loggers module composer.loggers.file_logger module composer.loggers.in_memory_logger module composer.loggers.logger_hparams module composer.loggers.registry module composer.loggers.tqdm_logger module composer.loggers.wandb_logger module composer.models module composer.models.base module composer.models.bert module composer.models.bert.bert_hparams module composer.models.bert.model module composer.models.classify_mnist module composer.models.classify_mnist.mnist_hparams module composer.models.classify_mnist.model module composer.models.deeplabv3 module composer.models.deeplabv3.deeplabv3 module composer.models.deeplabv3.deeplabv3_hparams module composer.models.efficientnetb0 module composer.models.efficientnetb0.efficientnetb0_hparams module composer.models.efficientnetb0.model module composer.models.efficientnets module composer.models.gpt2 module composer.models.gpt2.gpt2_hparams module composer.models.gpt2.model module composer.models.initializers module composer.models.loss module composer.models.model_hparams module composer.models.nlp_metrics module composer.models.resnet module composer.models.resnet.model module composer.models.resnet.resnet_hparams module composer.models.resnet20_cifar10 module composer.models.resnet20_cifar10.model module composer.models.resnet20_cifar10.resnet20_cifar10_hparams module composer.models.resnet56_cifar10 module composer.models.resnet56_cifar10.model module composer.models.resnet56_cifar10.resnet56_cifar10_hparams module composer.models.resnet9_cifar10 module composer.models.resnet9_cifar10.model module composer.models.resnet9_cifar10.resnet9_cifar10_hparams module composer.models.resnets module composer.models.timm module composer.models.timm.model module composer.models.timm.timm_hparams module composer.models.transformer_hparams module composer.models.transformer_shared module composer.models.unet module composer.models.unet.layers module composer.models.unet.model module composer.models.unet.unet module composer.models.unet.unet_hparams module composer.optim module composer.optim.decoupled_weight_decay module composer.optim.optimizer_hparams module composer.optim.scheduler module composer.profiler module composer.profiler.dataloader_profiler module composer.profiler.json_trace module composer.profiler.json_trace_merger module composer.profiler.profiler_hparams module composer.profiler.system_profiler module composer.profiler.torch_profiler module composer.trainer module composer.trainer.ddp module composer.trainer.devices module composer.trainer.devices.device module composer.trainer.devices.device_cpu module composer.trainer.devices.device_gpu module composer.trainer.devices.device_hparams module composer.trainer.trainer module composer.trainer.trainer_hparams module composer.utils module composer.utils.dist module composer.utils.iter_helpers module composer.utils.module_surgery module composer.utils.object_store module composer.utils.reproducibility module composer.utils.run_directory module composer.utils.string_enum module ComposerClassifier (class in composer.models.base) ComposerDeepLabV3 (class in composer.models.deeplabv3.deeplabv3) ComposerModel (class in composer.models.base) ComposerResNet (class in composer.models.resnet.model) ComposerSchedulerFn (class in composer.optim.scheduler) ComposerTransformer (class in composer.models.transformer_shared) compute() (composer.models.loss.CrossEntropyLoss method) (composer.models.loss.Dice method) (composer.models.loss.MIoU method) (composer.models.nlp_metrics.BinaryF1Score method) (composer.models.nlp_metrics.CrossEntropyLoss method) (composer.models.nlp_metrics.LanguageCrossEntropyLoss method) (composer.models.nlp_metrics.Perplexity method) configure_deterministic_mode() (in module composer.utils.reproducibility) constant_scheduler() (in module composer.optim.scheduler) ConstantLRHparams (class in composer.optim.scheduler) container_name (composer.utils.object_store.ObjectStoreProvider property) contrast() (in module composer.algorithms.utils.augmentation_primitives) contrast_original() (in module composer.algorithms.utils.augmentation_primitives) cosine_annealing_scheduler() (in module composer.optim.scheduler) cosine_annealing_warm_restarts_scheduler() (in module composer.optim.scheduler) cosine_annealing_with_warmup_scheduler() (in module composer.optim.scheduler) CosineAnnealingLRHparams (class in composer.optim.scheduler) CosineAnnealingWarmRestartsHparams (class in composer.optim.scheduler) CosineAnnealingWithWarmupLRHparams (class in composer.optim.scheduler) count_module_instances() (in module composer.utils.module_surgery) counter() (composer.profiler.Marker method) CPUDeviceHparams (class in composer.trainer.devices.device_hparams) CrossEntropyLoss (class in composer.models.loss) (class in composer.models.nlp_metrics) CutMix (class in composer.algorithms.cutmix.cutmix) cutmix_batch() (in module composer.algorithms.cutmix.cutmix) CutMixHparams (class in composer.algorithms.hparams) CutOut (class in composer.algorithms.cutout.cutout) cutout_batch() (in module composer.algorithms.cutout.cutout) CutOutHparams (class in composer.algorithms.hparams) D data (composer.loggers.in_memory_logger.InMemoryLogger attribute) DataLoader (class in composer.core.types) dataloader (composer.core.evaluator.Evaluator attribute) DataloaderHparams (class in composer.datasets.dataloader) DataloaderProfiler (class in composer.profiler.dataloader_profiler) dataset (composer.core.types.DataLoader attribute) DatasetHparams (class in composer.datasets.hparams) DataSpec (class in composer.core.data_spec) DDPSyncStrategy (class in composer.trainer.ddp) DecoupledAdamW (class in composer.optim.decoupled_weight_decay) DecoupledAdamWHparams (class in composer.optim.optimizer_hparams) DecoupledSGDW (class in composer.optim.decoupled_weight_decay) DecoupledSGDWHparams (class in composer.optim.optimizer_hparams) deeplabv3_builder() (in module composer.models.deeplabv3.deeplabv3) DeepLabV3Hparams (class in composer.models.deeplabv3.deeplabv3_hparams) deepspeed_enabled (composer.trainer.trainer.Trainer property) DepthwiseSeparableConv (class in composer.models.efficientnets) Device (class in composer.trainer.devices.device) DeviceCPU (class in composer.trainer.devices.device_cpu) DeviceGPU (class in composer.trainer.devices.device_gpu) DeviceHparams (class in composer.trainer.devices.device_hparams) Dice (class in composer.models.loss) dist_backend (composer.trainer.devices.device.Device attribute) download_object() (composer.utils.object_store.ObjectStoreProvider method) download_object_as_stream() (composer.utils.object_store.ObjectStoreProvider method) drop_connect() (in module composer.models.efficientnets) drop_last (composer.core.types.DataLoader attribute) DURATION (composer.core.time.TimeUnit attribute) E EfficientNet (class in composer.models.efficientnets) EfficientNetB0 (class in composer.models.efficientnetb0.model) EfficientNetB0Hparams (class in composer.models.efficientnetb0.efficientnetb0_hparams) Engine (class in composer.core.engine) engine (composer.trainer.trainer.Trainer attribute) ensure_tuple() (in module composer.utils.iter_helpers) EPOCH (composer.core.logging.logger.LogLevel attribute) epoch (composer.core.state.State property) (composer.core.time.Timer property) EPOCH (composer.core.time.TimeUnit attribute) EPOCH_END (composer.core.event.Event attribute) epoch_end() (composer.core.callback.Callback method) EPOCH_START (composer.core.event.Event attribute) epoch_start() (composer.core.callback.Callback method) equalize() (in module composer.algorithms.utils.augmentation_primitives) eval() (composer.trainer.trainer.Trainer method) EVAL_AFTER_FORWARD (composer.core.event.Event attribute) eval_after_forward() (composer.core.callback.Callback method) EVAL_BATCH_END (composer.core.event.Event attribute) eval_batch_end() (composer.core.callback.Callback method) EVAL_BATCH_START (composer.core.event.Event attribute) eval_batch_start() (composer.core.callback.Callback method) EVAL_BEFORE_FORWARD (composer.core.event.Event attribute) eval_before_forward() (composer.core.callback.Callback method) EVAL_END (composer.core.event.Event attribute) eval_end() (composer.core.callback.Callback method) EVAL_START (composer.core.event.Event attribute) eval_start() (composer.core.callback.Callback method) Evaluator (class in composer.core.evaluator) EvaluatorHparams (class in composer.datasets.evaluator) evaluators (composer.trainer.trainer.Trainer attribute) Event (class in composer.core.event) event_handlers (composer.profiler.Profiler property) exit_code (composer.core.engine.Trace attribute) exponential_scheduler() (in module composer.optim.scheduler) ExponentialLRHparams (class in composer.optim.scheduler) F Factorize (class in composer.algorithms.factorize.factorize) factorize_conv2d() (in module composer.algorithms.factorize.factorize_core) factorize_matrix() (in module composer.algorithms.factorize.factorize_core) FactorizedConv2d (class in composer.algorithms.factorize.factorize_modules) FactorizedLinear (class in composer.algorithms.factorize.factorize_modules) FactorizeHparams (class in composer.algorithms.hparams) factorizing_could_speedup() (in module composer.algorithms.factorize.factorize_modules) FileLogger (class in composer.loggers.file_logger) FileLoggerHparams (class in composer.loggers.logger_hparams) find_unused_parameters (composer.algorithms.layer_freezing.layer_freezing.LayerFreezing property) (composer.algorithms.stochastic_depth.stochastic_depth.StochasticDepth property) (composer.core.algorithm.Algorithm property) finish() (composer.profiler.Marker method) FIT (composer.core.logging.logger.LogLevel attribute) fit() (composer.trainer.trainer.Trainer method) FIT_START (composer.core.event.Event attribute) fit_start() (composer.core.callback.Callback method) FORCED_SYNC (composer.trainer.ddp.DDPSyncStrategy attribute) format_log_data_value() (in module composer.core.logging.logger) FP16 (composer.core.precision.Precision attribute) FP32 (composer.core.precision.Precision attribute) freeze_layers() (in module composer.algorithms.layer_freezing.layer_freezing) from_batch() (composer.core.time.Time class method) from_duration() (composer.core.time.Time class method) from_epoch() (composer.core.time.Time class method) from_sample() (composer.core.time.Time class method) from_target_layer() (composer.algorithms.stochastic_depth.sample_stochastic_layers.SampleStochasticBottleneck static method) (composer.algorithms.stochastic_depth.stochastic_layers.StochasticBottleneck static method) from_timestring() (composer.core.time.Time class method) from_token() (composer.core.time.Time class method) G get() (composer.core.time.Timer method) get_action() (composer.profiler.Profiler method) get_elapsed_duration() (composer.core.state.State method) get_global_rank() (in module composer.utils.dist) get_local_rank() (in module composer.utils.dist) get_local_world_size() (in module composer.utils.dist) get_logger_hparams() (in module composer.loggers.registry) get_model_from_name() (composer.models.efficientnets.EfficientNet static method) (composer.models.resnets.CIFAR_ResNet static method) get_model_inputs() (composer.models.transformer_shared.ComposerTransformer method) get_modified_files() (in module composer.utils.run_directory) get_node_rank() (in module composer.utils.dist) get_node_run_directory() (in module composer.utils.run_directory) get_object_size() (composer.utils.object_store.ObjectStoreProvider method) get_optimizer() (in module composer.optim.optimizer_hparams) get_random_seed() (in module composer.utils.reproducibility) get_run_directory() (in module composer.utils.run_directory) get_run_directory_timestamp() (in module composer.utils.run_directory) get_sampler() (in module composer.utils.dist) get_timestamp() (composer.core.time.Timer method) get_uri_for_uploaded_file() (composer.callbacks.run_directory_uploader.RunDirectoryUploader method) get_world_size() (in module composer.utils.dist) GhostBatchNorm (class in composer.algorithms.ghost_batchnorm.ghost_batchnorm) GhostBatchNormHparams (class in composer.algorithms.hparams) GLUEHparams (class in composer.datasets.glue) GPT2Hparams (class in composer.models.gpt2.gpt2_hparams) GPT2Model (class in composer.models.gpt2.model) GPUDeviceHparams (class in composer.trainer.devices.device_hparams) GradMonitor (class in composer.callbacks.grad_monitor) GradMonitorHparams (class in composer.callbacks.callback_hparams) I image_as_type() (in module composer.algorithms.utils.augmentation_common) ImagenetDatasetHparams (class in composer.datasets.imagenet) in_channels (composer.algorithms.factorize.factorize_modules.FactorizedConv2d property) in_features (composer.algorithms.factorize.factorize_modules.FactorizedLinear property) inference2d() (composer.models.unet.unet.UNet method) INIT (composer.core.event.Event attribute) init() (composer.core.callback.Callback method) initialize_dist() (in module composer.utils.dist) initialize_object() (composer.algorithms.algorithm_hparams.AlgorithmHparams method) (composer.callbacks.callback_hparams.CallbackHparams method) (composer.callbacks.callback_hparams.GradMonitorHparams method) (composer.callbacks.callback_hparams.LRMonitorHparams method) (composer.callbacks.callback_hparams.MemoryMonitorHparams method) (composer.callbacks.callback_hparams.RunDirectoryUploaderHparams method) (composer.callbacks.callback_hparams.SpeedMonitorHparams method) (composer.datasets.ade20k.ADE20kDatasetHparams method) (composer.datasets.brats.BratsDatasetHparams method) (composer.datasets.cifar10.CIFAR10DatasetHparams method) (composer.datasets.dataloader.DataloaderHparams method) (composer.datasets.evaluator.EvaluatorHparams method) (composer.datasets.glue.GLUEHparams method) (composer.datasets.hparams.DatasetHparams method) (composer.datasets.imagenet.ImagenetDatasetHparams method) (composer.datasets.lm_datasets.LMDatasetHparams method) (composer.datasets.mnist.MNISTDatasetHparams method) (composer.loggers.logger_hparams.LoggerCallbackHparams method) (composer.loggers.logger_hparams.WandBLoggerHparams method) (composer.models.model_hparams.ModelHparams method) (composer.profiler.profiler_hparams.ProfilerEventHandlerHparams method) (composer.utils.object_store.ObjectStoreProviderHparams method) Initializer (class in composer.models.model_hparams) InMemoryLogger (class in composer.loggers.in_memory_logger) InMemoryLoggerHparams (class in composer.loggers.logger_hparams) instant() (composer.profiler.Marker method) is_after_event (composer.core.event.Event property) is_available() (in module composer.utils.dist) is_before_event (composer.core.event.Event property) is_initialized() (in module composer.utils.dist) iterate_with_pbar() (in module composer.utils.iter_helpers) J JSON (in module composer.core.types) JSONTraceHandler (class in composer.profiler.json_trace) JSONTraceHandlerHparams (class in composer.profiler.profiler_hparams) L label (composer.core.evaluator.Evaluator attribute) LabelSmoothing (class in composer.algorithms.label_smoothing.label_smoothing) LabelSmoothingHparams (class in composer.algorithms.hparams) LanguageCrossEntropyLoss (class in composer.models.nlp_metrics) latent_channels (composer.algorithms.factorize.factorize_modules.FactorizedConv2d property) latent_features (composer.algorithms.factorize.factorize_modules.FactorizedLinear property) LayerFreezing (class in composer.algorithms.layer_freezing.layer_freezing) LayerFreezingHparams (class in composer.algorithms.hparams) linear_scheduler() (in module composer.optim.scheduler) linear_with_warmup_scheduler() (in module composer.optim.scheduler) LinearLRHparams (class in composer.optim.scheduler) LinearWithWarmupLRHparams (class in composer.optim.scheduler) LMDatasetHparams (class in composer.datasets.lm_datasets) load_model_state() (composer.core.state.State method) load_state_dict() (composer.callbacks.speed_monitor.SpeedMonitor method) (composer.core.serializable.Serializable method) (composer.core.state.State method) log_metric() (composer.core.logging.base_backend.LoggerCallback method) Logger (class in composer.core.logging.logger) logger (composer.trainer.trainer.Trainer attribute) LoggerCallback (class in composer.core.logging.base_backend) LoggerCallbackHparams (class in composer.loggers.logger_hparams) LogLevel (class in composer.core.logging.logger) loss (composer.core.state.State attribute) loss() (composer.models.base.ComposerModel method) (composer.models.deeplabv3.deeplabv3.ComposerDeepLabV3 method) (composer.models.transformer_shared.ComposerTransformer method) LowRankSolution (class in composer.algorithms.factorize.factorize_core) LRMonitor (class in composer.callbacks.lr_monitor) LRMonitorHparams (class in composer.callbacks.callback_hparams) M map_collection() (in module composer.utils.iter_helpers) map_pillow_function() (in module composer.algorithms.utils.augmentation_common) Marker (class in composer.profiler) marker() (composer.profiler.Profiler method) MaskedAccuracy (class in composer.models.nlp_metrics) match() (composer.algorithms.blurpool.blurpool.BlurPool method) (composer.algorithms.channels_last.channels_last.ChannelsLast method) (composer.algorithms.cutmix.cutmix.CutMix method) (composer.algorithms.cutout.cutout.CutOut method) (composer.algorithms.factorize.factorize.Factorize method) (composer.algorithms.ghost_batchnorm.ghost_batchnorm.GhostBatchNorm method) (composer.algorithms.layer_freezing.layer_freezing.LayerFreezing method) (composer.algorithms.mixup.mixup.MixUp method) (composer.algorithms.progressive_resizing.progressive_resizing.ProgressiveResizing method) (composer.algorithms.sam.sam.SAM method) (composer.algorithms.scale_schedule.scale_schedule.ScaleSchedule method) (composer.algorithms.selective_backprop.selective_backprop.SelectiveBackprop method) (composer.algorithms.squeeze_excite.squeeze_excite.SqueezeExcite method) (composer.algorithms.stochastic_depth.stochastic_depth.StochasticDepth method) (composer.core.algorithm.Algorithm method) max_allowed_latent_channels() (composer.algorithms.factorize.factorize_modules.FactorizedLinear static method) max_allowed_latent_features() (composer.algorithms.factorize.factorize_modules.FactorizedConv2d static method) max_epochs (composer.core.state.State property) MBConvBlock (class in composer.models.efficientnets) MemoryFormat (class in composer.core.types) MemoryMonitor (class in composer.callbacks.memory_monitor) MemoryMonitorHparams (class in composer.callbacks.callback_hparams) merge_traces() (in module composer.profiler.json_trace_merger) metric() (composer.core.logging.logger.Logger method) metric_batch() (composer.core.logging.logger.Logger method) metric_epoch() (composer.core.logging.logger.Logger method) metric_fit() (composer.core.logging.logger.Logger method) metrics (composer.core.evaluator.Evaluator attribute) Metrics (in module composer.core.types) metrics() (composer.models.base.ComposerModel method) (composer.models.deeplabv3.deeplabv3.ComposerDeepLabV3 method) (composer.models.transformer_shared.ComposerTransformer method) MIoU (class in composer.models.loss) MixUp (class in composer.algorithms.mixup.mixup) mixup_batch() (in module composer.algorithms.mixup.mixup) MixUpHparams (class in composer.algorithms.hparams) MNIST_Classifier (class in composer.models.classify_mnist.model) MnistClassifierHparams (class in composer.models.classify_mnist.mnist_hparams) MNISTDatasetHparams (class in composer.datasets.mnist) Model (class in composer.models.classify_mnist.model) (in module composer.core.types) ModelHparams (class in composer.models.model_hparams) ModelParameters (in module composer.core.types) module composer.algorithms composer.algorithms.algorithm_hparams composer.algorithms.algorithm_registry composer.algorithms.alibi composer.algorithms.alibi.alibi composer.algorithms.augmix composer.algorithms.augmix.augmix composer.algorithms.blurpool composer.algorithms.blurpool.blurpool composer.algorithms.blurpool.blurpool_layers composer.algorithms.channels_last composer.algorithms.channels_last.channels_last composer.algorithms.colout composer.algorithms.colout.colout composer.algorithms.cutmix composer.algorithms.cutmix.cutmix composer.algorithms.cutout composer.algorithms.cutout.cutout composer.algorithms.factorize composer.algorithms.factorize.factorize composer.algorithms.factorize.factorize_core composer.algorithms.factorize.factorize_modules composer.algorithms.ghost_batchnorm composer.algorithms.ghost_batchnorm.ghost_batchnorm composer.algorithms.hparams composer.algorithms.label_smoothing composer.algorithms.label_smoothing.label_smoothing composer.algorithms.layer_freezing composer.algorithms.layer_freezing.layer_freezing composer.algorithms.mixup composer.algorithms.mixup.mixup composer.algorithms.no_op_model composer.algorithms.no_op_model.no_op_model composer.algorithms.progressive_resizing composer.algorithms.progressive_resizing.progressive_resizing composer.algorithms.randaugment composer.algorithms.randaugment.randaugment composer.algorithms.sam composer.algorithms.sam.sam composer.algorithms.scale_schedule composer.algorithms.scale_schedule.scale_schedule composer.algorithms.selective_backprop composer.algorithms.selective_backprop.selective_backprop composer.algorithms.seq_length_warmup composer.algorithms.seq_length_warmup.seq_length_warmup composer.algorithms.squeeze_excite composer.algorithms.squeeze_excite.squeeze_excite composer.algorithms.stochastic_depth composer.algorithms.stochastic_depth.sample_stochastic_layers composer.algorithms.stochastic_depth.stochastic_depth composer.algorithms.stochastic_depth.stochastic_layers composer.algorithms.swa composer.algorithms.swa.swa composer.algorithms.utils composer.algorithms.utils.augmentation_common composer.algorithms.utils.augmentation_primitives composer.callbacks composer.callbacks.callback_hparams composer.callbacks.grad_monitor composer.callbacks.lr_monitor composer.callbacks.memory_monitor composer.callbacks.run_directory_uploader composer.callbacks.speed_monitor composer.core composer.core.algorithm composer.core.callback composer.core.data_spec composer.core.engine composer.core.evaluator composer.core.event composer.core.logging composer.core.logging.base_backend composer.core.logging.logger composer.core.precision composer.core.serializable composer.core.state composer.core.time composer.core.types composer.datasets composer.datasets.ade20k composer.datasets.brats composer.datasets.cifar10 composer.datasets.dataloader composer.datasets.dataset_registry composer.datasets.evaluator composer.datasets.glue composer.datasets.hparams composer.datasets.imagenet composer.datasets.lm_datasets composer.datasets.mnist composer.datasets.synthetic composer.datasets.utils composer.functional composer.loggers composer.loggers.file_logger composer.loggers.in_memory_logger composer.loggers.logger_hparams composer.loggers.registry composer.loggers.tqdm_logger composer.loggers.wandb_logger composer.models composer.models.base composer.models.bert composer.models.bert.bert_hparams composer.models.bert.model composer.models.classify_mnist composer.models.classify_mnist.mnist_hparams composer.models.classify_mnist.model composer.models.deeplabv3 composer.models.deeplabv3.deeplabv3 composer.models.deeplabv3.deeplabv3_hparams composer.models.efficientnetb0 composer.models.efficientnetb0.efficientnetb0_hparams composer.models.efficientnetb0.model composer.models.efficientnets composer.models.gpt2 composer.models.gpt2.gpt2_hparams composer.models.gpt2.model composer.models.initializers composer.models.loss composer.models.model_hparams composer.models.nlp_metrics composer.models.resnet composer.models.resnet.model composer.models.resnet.resnet_hparams composer.models.resnet20_cifar10 composer.models.resnet20_cifar10.model composer.models.resnet20_cifar10.resnet20_cifar10_hparams composer.models.resnet56_cifar10 composer.models.resnet56_cifar10.model composer.models.resnet56_cifar10.resnet56_cifar10_hparams composer.models.resnet9_cifar10 composer.models.resnet9_cifar10.model composer.models.resnet9_cifar10.resnet9_cifar10_hparams composer.models.resnets composer.models.timm composer.models.timm.model composer.models.timm.timm_hparams composer.models.transformer_hparams composer.models.transformer_shared composer.models.unet composer.models.unet.layers composer.models.unet.model composer.models.unet.unet composer.models.unet.unet_hparams composer.optim composer.optim.decoupled_weight_decay composer.optim.optimizer_hparams composer.optim.scheduler composer.profiler composer.profiler.dataloader_profiler composer.profiler.json_trace composer.profiler.json_trace_merger composer.profiler.profiler_hparams composer.profiler.system_profiler composer.profiler.torch_profiler composer.trainer composer.trainer.ddp composer.trainer.devices composer.trainer.devices.device composer.trainer.devices.device_cpu composer.trainer.devices.device_gpu composer.trainer.devices.device_hparams composer.trainer.trainer composer.trainer.trainer_hparams composer.utils composer.utils.dist composer.utils.iter_helpers composer.utils.module_surgery composer.utils.object_store composer.utils.reproducibility composer.utils.run_directory composer.utils.string_enum module_to_device() (composer.trainer.devices.device.Device method) most_recent_timestamps (composer.loggers.in_memory_logger.InMemoryLogger attribute) most_recent_values (composer.loggers.in_memory_logger.InMemoryLogger attribute) MULTI_AUTO_SYNC (composer.trainer.ddp.DDPSyncStrategy attribute) multi_step_scheduler() (in module composer.optim.scheduler) multi_step_with_warmup_scheduler() (in module composer.optim.scheduler) MultiStepLRHparams (class in composer.optim.scheduler) MultiStepWithWarmupLRHparams (class in composer.optim.scheduler) N NoOpModelHparams (class in composer.algorithms.hparams) NormalizationFn (class in composer.datasets.utils) num_workers (composer.core.types.DataLoader attribute) O ObjectStoreProvider (class in composer.utils.object_store) ObjectStoreProviderHparams (class in composer.utils.object_store) on_batch_complete() (composer.core.time.Timer method) on_epoch_complete() (composer.core.time.Timer method) Optimizer (in module composer.core.types) optimizer_to_device() (composer.trainer.devices.device.Device method) OptimizerHparams (class in composer.optim.optimizer_hparams) Optimizers (in module composer.core.types) order (composer.core.engine.Trace attribute) out_channels (composer.algorithms.factorize.factorize_modules.FactorizedConv2d property) out_features (composer.algorithms.factorize.factorize_modules.FactorizedLinear property) outputs (composer.core.state.State attribute) P PadToSize (class in composer.datasets.ade20k) Perplexity (class in composer.models.nlp_metrics) PhotometricDistoration (class in composer.datasets.ade20k) pil_image_collate() (in module composer.datasets.utils) pin_memory (composer.core.types.DataLoader attribute) polynomial_scheduler() (in module composer.optim.scheduler) PolynomialLRHparams (class in composer.optim.scheduler) post_close() (composer.core.callback.Callback method) posterize() (in module composer.algorithms.utils.augmentation_primitives) Precision (class in composer.core.precision) precision (composer.core.state.State property) precision_context() (composer.trainer.devices.device.Device method) prefetch_factor (composer.core.types.DataLoader attribute) process_counter_event() (composer.profiler.ProfilerEventHandler method) process_duration_event() (composer.profiler.ProfilerEventHandler method) process_instant_event() (composer.profiler.ProfilerEventHandler method) Profiler (class in composer.profiler) ProfilerAction (class in composer.profiler) ProfilerEventHandler (class in composer.profiler) ProfilerEventHandlerHparams (class in composer.profiler.profiler_hparams) ProgressiveResizing (class in composer.algorithms.progressive_resizing.progressive_resizing) ProgressiveResizingHparams (class in composer.algorithms.hparams) provider_name (composer.utils.object_store.ObjectStoreProvider property) R RAdamHparams (class in composer.optim.optimizer_hparams) RandAugment (class in composer.algorithms.randaugment.randaugment) randaugment_image() (in module composer.algorithms.randaugment.randaugment) RandAugmentHparams (class in composer.algorithms.hparams) RandAugmentTransform (class in composer.algorithms.randaugment.randaugment) RandomCropPair (class in composer.datasets.ade20k) RandomHFlipPair (class in composer.datasets.ade20k) RandomResizePair (class in composer.datasets.ade20k) replace_module_classes() (in module composer.utils.module_surgery) replace_params_in_optimizer() (in module composer.utils.module_surgery) ReplacementFunction (in module composer.utils.module_surgery) resize_batch() (in module composer.algorithms.progressive_resizing.progressive_resizing) ResNet9 (class in composer.models.resnet9_cifar10.model) ResNetHparams (class in composer.models.resnet.resnet_hparams) RMSPropHparams (class in composer.optim.optimizer_hparams) rotate() (in module composer.algorithms.utils.augmentation_primitives) round_channels() (in module composer.models.efficientnets) run (composer.core.engine.Trace attribute) run_event() (composer.core.callback.Callback method) (composer.core.engine.Engine method) RunDirectoryUploader (class in composer.callbacks.run_directory_uploader) RunDirectoryUploaderHparams (class in composer.callbacks.callback_hparams) S SAM (class in composer.algorithms.sam.sam) SAMHparams (class in composer.algorithms.hparams) SAMOptimizer (class in composer.algorithms.sam.sam) sample (composer.core.time.Timer property) SAMPLE (composer.core.time.TimeUnit attribute) sample_in_epoch (composer.core.time.Timer property) sampler (composer.core.types.DataLoader attribute) SampleStochasticBottleneck (class in composer.algorithms.stochastic_depth.sample_stochastic_layers) saved_checkpoints (composer.trainer.trainer.Trainer property) Scaler (in module composer.core.types) ScaleSchedule (class in composer.algorithms.scale_schedule.scale_schedule) ScaleScheduleHparams (class in composer.algorithms.hparams) Scheduler (in module composer.core.types) scheduler_function() (composer.optim.scheduler.ConstantLRHparams method) (composer.optim.scheduler.CosineAnnealingLRHparams method) (composer.optim.scheduler.CosineAnnealingWarmRestartsHparams method) (composer.optim.scheduler.CosineAnnealingWithWarmupLRHparams method) (composer.optim.scheduler.ExponentialLRHparams method) (composer.optim.scheduler.LinearLRHparams method) (composer.optim.scheduler.LinearWithWarmupLRHparams method) (composer.optim.scheduler.MultiStepLRHparams method) (composer.optim.scheduler.MultiStepWithWarmupLRHparams method) (composer.optim.scheduler.PolynomialLRHparams method) (composer.optim.scheduler.StepLRHparams method) SchedulerHparams (class in composer.optim.scheduler) Schedulers (in module composer.core.types) seed_all() (in module composer.utils.reproducibility) select_using_loss() (in module composer.algorithms.selective_backprop.selective_backprop) SelectiveBackprop (class in composer.algorithms.selective_backprop.selective_backprop) SelectiveBackpropHparams (class in composer.algorithms.hparams) SeqLengthWarmup (class in composer.algorithms.seq_length_warmup) (class in composer.algorithms.seq_length_warmup.seq_length_warmup) SeqLengthWarmupHparams (class in composer.algorithms.hparams) Serializable (class in composer.core.serializable) serialized_attributes (composer.core.state.State attribute) set_batch_sequence_length (class in composer.algorithms.seq_length_warmup) set_batch_sequence_length() (in module composer.algorithms.seq_length_warmup.seq_length_warmup) SGDHparams (class in composer.optim.optimizer_hparams) sgdw() (composer.optim.decoupled_weight_decay.DecoupledSGDW static method) sharpness() (in module composer.algorithms.utils.augmentation_primitives) sharpness_original() (in module composer.algorithms.utils.augmentation_primitives) shear_x() (in module composer.algorithms.utils.augmentation_primitives) shear_y() (in module composer.algorithms.utils.augmentation_primitives) should_selective_backprop() (in module composer.algorithms.selective_backprop.selective_backprop) SINGLE_AUTO_SYNC (composer.trainer.ddp.DDPSyncStrategy attribute) SKIP (composer.profiler.ProfilerAction attribute) smooth_labels() (in module composer.algorithms.label_smoothing.label_smoothing) soft_cross_entropy() (in module composer.models.loss) solarize() (in module composer.algorithms.utils.augmentation_primitives) SpeedMonitor (class in composer.callbacks.speed_monitor) SpeedMonitorHparams (class in composer.callbacks.callback_hparams) SqueezeExcite (class in composer.algorithms.squeeze_excite.squeeze_excite) (class in composer.models.efficientnets) SqueezeExcite2d (class in composer.algorithms.squeeze_excite.squeeze_excite) SqueezeExciteConv2d (class in composer.algorithms.squeeze_excite.squeeze_excite) SqueezeExciteHparams (class in composer.algorithms.hparams) start() (composer.profiler.Marker method) State (class in composer.core.state) state (composer.trainer.trainer.Trainer attribute) state_dict() (composer.callbacks.speed_monitor.SpeedMonitor method) (composer.core.serializable.Serializable method) (composer.core.state.State method) step (composer.core.state.State property) step() (composer.optim.decoupled_weight_decay.DecoupledAdamW method) (composer.optim.decoupled_weight_decay.DecoupledSGDW method) step_scheduler() (in module composer.optim.scheduler) StepLRHparams (class in composer.optim.scheduler) steps_per_epoch (composer.core.state.State property) StochasticBottleneck (class in composer.algorithms.stochastic_depth.stochastic_layers) StochasticDepth (class in composer.algorithms.stochastic_depth.stochastic_depth) StochasticDepthHparams (class in composer.algorithms.hparams) StringEnum (class in composer.utils.string_enum) SWA (class in composer.algorithms.swa.swa) SWAHparams (class in composer.algorithms.hparams) SyntheticBatchPairDataset (class in composer.datasets.synthetic) SyntheticDataLabelType (class in composer.datasets.synthetic) SyntheticDataType (class in composer.datasets.synthetic) SyntheticHparamsMixin (class in composer.datasets.hparams) SyntheticPILDataset (class in composer.datasets.synthetic) SystemProfiler (class in composer.profiler.system_profiler) T tensor_to_device() (composer.trainer.devices.device.Device method) Tensors (in module composer.core.types) Time (class in composer.core.time) timeout (composer.core.types.DataLoader attribute) Timer (class in composer.core.time) timer (composer.core.state.State attribute) Timestamp (class in composer.core.time) TimeUnit (class in composer.core.time) Timm (class in composer.models.timm.model) TimmHparams (class in composer.models.timm.timm_hparams) TLogData (in module composer.core.logging.logger) TLogDataValue (in module composer.core.logging.logger) token (composer.core.time.Timer property) TOKEN (composer.core.time.TimeUnit attribute) token_in_epoch (composer.core.time.Timer property) TorchProfiler (class in composer.profiler.torch_profiler) TQDMLogger (class in composer.loggers.tqdm_logger) TQDMLoggerHparams (class in composer.loggers.logger_hparams) Trace (class in composer.core.engine) Traces (in module composer.core.engine) Trainer (class in composer.trainer.trainer) TrainerHparams (class in composer.trainer.trainer_hparams) TransformerHparams (class in composer.models.transformer_hparams) translate_x() (in module composer.algorithms.utils.augmentation_primitives) translate_y() (in module composer.algorithms.utils.augmentation_primitives) U UNet (class in composer.models.unet.unet) UnetHparams (class in composer.models.unet.unet_hparams) unit (composer.core.time.Time property) unwrap_data_loader() (in module composer.datasets.dataloader) update() (composer.models.loss.CrossEntropyLoss method) (composer.models.loss.Dice method) (composer.models.loss.MIoU method) (composer.models.nlp_metrics.BinaryF1Score method) (composer.models.nlp_metrics.CrossEntropyLoss method) (composer.models.nlp_metrics.LanguageCrossEntropyLoss method) update_params_in_optimizer() (in module composer.utils.module_surgery) upload_object() (composer.utils.object_store.ObjectStoreProvider method) upload_object_via_stream() (composer.utils.object_store.ObjectStoreProvider method) V validate() (composer.datasets.ade20k.ADE20kDatasetHparams method) (composer.datasets.glue.GLUEHparams method) (composer.datasets.lm_datasets.LMDatasetHparams method) (composer.models.base.ComposerModel method) (composer.models.bert.model.BERTModel method) (composer.models.deeplabv3.deeplabv3.ComposerDeepLabV3 method) (composer.models.transformer_shared.ComposerTransformer method) value (composer.core.time.Time property) W WandBLogger (class in composer.loggers.wandb_logger) WandBLoggerHparams (class in composer.loggers.logger_hparams) WARMUP (composer.profiler.ProfilerAction attribute) will_log() (composer.core.logging.base_backend.LoggerCallback method)