Index _ | A | B | C | D | E | F | G | H | I | J | L | M | N | O | P | R | S | T | U | V | W | Z _ __call__() (composer.optim.scheduler.ComposerScheduler method) A ACTIVE (composer.profiler.profiler_action.ProfilerAction attribute) ACTIVE_AND_SAVE (composer.profiler.profiler_action.ProfilerAction attribute) AdamHparams (class in composer.optim.optimizer_hparams_registry) adamw() (composer.optim.decoupled_weight_decay.DecoupledAdamW static method) AdamWHparams (class in composer.optim.optimizer_hparams_registry) add_vision_dataset_transform() (in module composer.datasets.utils) ADE20k (class in composer.datasets.ade20k) ADE20kDatasetHparams (class in composer.datasets.ade20k_hparams) 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) algorithms (composer.core.state.State property) Alibi (class in composer.algorithms.alibi.alibi) 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.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_fused_layernorm() (in module composer.algorithms.fused_layernorm.fused_layernorm) apply_gated_linear_units() (in module composer.algorithms.gated_linear_units.gated_linear_units) apply_ghost_batchnorm() (in module composer.algorithms.ghost_batchnorm.ghost_batchnorm) apply_gradient_clipping() (in module composer.algorithms.gradient_clipping.gradient_clipping) apply_squeeze_excite() (in module composer.algorithms.squeeze_excite.squeeze_excite) apply_stochastic_depth() (in module composer.algorithms.stochastic_depth.stochastic_depth) apply_stochastic_residual() (in module composer.utils.fx_utils) artifact_name (composer.loggers.file_logger.FileLogger property) 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) autocontrast() (in module composer.algorithms.utils.augmentation_primitives) B backwards_create_graph (composer.core.algorithm.Algorithm property) barrier() (in module composer.utils.dist) batch (composer.core.state.State attribute) (composer.core.time.Timestamp property) BATCH (composer.core.time.TimeUnit attribute) (composer.loggers.logger.LogLevel attribute) Batch (in module composer.core.types) BATCH_CHECKPOINT (composer.core.event.Event attribute) batch_checkpoint() (composer.core.callback.Callback method) BATCH_END (composer.core.event.Event attribute) batch_end() (composer.core.callback.Callback method) batch_get() (in module composer.utils.batch_helpers) batch_get_item() (composer.core.state.State method) batch_in_epoch (composer.core.time.Timestamp property) batch_set() (in module composer.utils.batch_helpers) batch_set_item() (composer.core.state.State method) BATCH_START (composer.core.event.Event attribute) batch_start() (composer.core.callback.Callback method) batch_to_device() (composer.trainer.devices.device.Device method) batch_wct (composer.core.time.Timestamp property) BEFORE_BACKWARD (composer.core.event.Event attribute) before_backward() (composer.core.callback.Callback method) BEFORE_DATALOADER (composer.core.event.Event attribute) before_dataloader() (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) BERTGatedFFOutput (class in composer.algorithms.gated_linear_units.gated_linear_unit_layers) BERTHparams (class in composer.models.bert.bert_hparams) BF16 (composer.core.precision.Precision attribute) bind_to_state() (composer.profiler.profiler.Profiler method) block_stochastic_forward() (in module composer.algorithms.stochastic_depth.stochastic_layers) BlockStochasticModule (class in composer.algorithms.stochastic_depth.stochastic_layers) 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) BratsDatasetHparams (class in composer.datasets.brats_hparams) 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) bytes_to_sample_dict() (in module composer.datasets.streaming.format) C C4Dataset (class in composer.datasets.c4) C4DatasetHparams (class in composer.datasets.c4_hparams) calc_iou_tensor() (in module composer.models.ssd.utils) Callback (class in composer.core.callback) callbacks (composer.core.state.State property) can_log_file_artifacts() (composer.loggers.logger_destination.LoggerDestination method) (composer.loggers.object_store_logger.ObjectStoreLogger method) (composer.loggers.wandb_logger.WandBLogger method) canonical_name (composer.core.event.Event property) CHANNELS_LAST (composer.core.types.MemoryFormat attribute) CHANNELS_LAST_3D (composer.core.types.MemoryFormat attribute) ChannelsLast (class in composer.algorithms.channels_last.channels_last) checkpoint_periodically() (in module composer.callbacks.checkpoint_saver) CheckpointSaver (class in composer.callbacks.checkpoint_saver) CIFAR10DatasetHparams (class in composer.datasets.cifar_hparams) CLASSIFICATION_INT (composer.datasets.synthetic.SyntheticDataLabelType attribute) CLASSIFICATION_ONE_HOT (composer.datasets.synthetic.SyntheticDataLabelType attribute) close() (composer.core.callback.Callback method) (composer.core.engine.Engine method) (composer.trainer.trainer.Trainer method) (composer.utils.object_store.object_store.ObjectStore method) COCODatasetHparams (class in composer.datasets.coco_hparams) COCODetection (class in composer.datasets.coco) 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) ColOutTransform (class in composer.algorithms.colout.colout) compile_composer_scheduler() (in module composer.optim.scheduler) composer.algorithms module composer.algorithms.algorithm_hparams_registry module composer.algorithms.alibi module composer.algorithms.alibi.alibi module composer.algorithms.alibi.attention_surgery_functions module composer.algorithms.alibi.attention_surgery_functions.utils 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.ema module composer.algorithms.ema.ema module composer.algorithms.factorize module composer.algorithms.factorize.factorize module composer.algorithms.factorize.factorize_core module composer.algorithms.factorize.factorize_modules module composer.algorithms.fused_layernorm module composer.algorithms.fused_layernorm.fused_layernorm module composer.algorithms.gated_linear_units module composer.algorithms.gated_linear_units.gated_linear_unit_layers module composer.algorithms.gated_linear_units.gated_linear_units module composer.algorithms.ghost_batchnorm module composer.algorithms.ghost_batchnorm.ghost_batchnorm module composer.algorithms.gradient_clipping module composer.algorithms.gradient_clipping.gradient_clipping 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.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.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.algorithms.warnings module composer.callbacks module composer.callbacks.callback_hparams_registry module composer.callbacks.checkpoint_saver module composer.callbacks.early_stopper module composer.callbacks.export_for_inference module composer.callbacks.grad_monitor module composer.callbacks.image_visualizer module composer.callbacks.lr_monitor module composer.callbacks.memory_monitor module composer.callbacks.mlperf module composer.callbacks.speed_monitor module composer.callbacks.threshold_stopper 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.passes 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.ade20k_hparams module composer.datasets.brats module composer.datasets.brats_hparams module composer.datasets.c4 module composer.datasets.c4_hparams module composer.datasets.cifar module composer.datasets.cifar_hparams module composer.datasets.coco module composer.datasets.coco_hparams module composer.datasets.dataset_hparams module composer.datasets.dataset_hparams_registry module composer.datasets.evaluator_hparams module composer.datasets.ffcv_utils module composer.datasets.glue_hparams module composer.datasets.imagenet module composer.datasets.imagenet_hparams module composer.datasets.lm_dataset_hparams module composer.datasets.mnist_hparams module composer.datasets.streaming module composer.datasets.streaming.dataset module composer.datasets.streaming.download module composer.datasets.streaming.format module composer.datasets.streaming.world module composer.datasets.streaming.writer module composer.datasets.synthetic module composer.datasets.synthetic_hparams module composer.datasets.synthetic_lm 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 module composer.loggers.logger_destination module composer.loggers.logger_hparams_registry module composer.loggers.object_store_logger module composer.loggers.progress_bar_logger module composer.loggers.tensorboard_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_hparams module composer.models.deeplabv3.model module composer.models.efficientnetb0 module composer.models.efficientnetb0.efficientnetb0_hparams module composer.models.efficientnetb0.efficientnets module composer.models.efficientnetb0.model module composer.models.gpt2 module composer.models.gpt2.gpt2_hparams module composer.models.gpt2.model module composer.models.huggingface module composer.models.initializers module composer.models.model_hparams module composer.models.resnet module composer.models.resnet.model module composer.models.resnet.resnet_hparams module composer.models.resnet_cifar module composer.models.resnet_cifar.model module composer.models.resnet_cifar.resnet_cifar_hparams module composer.models.resnet_cifar.resnets module composer.models.ssd module composer.models.ssd.base_model module composer.models.ssd.ssd module composer.models.ssd.ssd300 module composer.models.ssd.ssd_hparams module composer.models.ssd.utils module composer.models.tasks module composer.models.tasks.classification module composer.models.timm module composer.models.timm.model module composer.models.timm.timm_hparams module composer.models.unet 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_registry module composer.optim.scheduler module composer.optim.scheduler_hparams_registry module composer.profiler module composer.profiler.json_trace_handler module composer.profiler.json_trace_merger module composer.profiler.marker module composer.profiler.profiler module composer.profiler.profiler_action module composer.profiler.profiler_hparams module composer.profiler.profiler_schedule module composer.profiler.system_profiler module composer.profiler.torch_profiler module composer.profiler.trace_handler 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_registry module composer.trainer.devices.device_mps module composer.trainer.devices.device_tpu module composer.trainer.trainer module composer.trainer.trainer_hparams module composer.utils module composer.utils.batch_helpers module composer.utils.checkpoint module composer.utils.collect_env module composer.utils.dist module composer.utils.file_helpers module composer.utils.fx_utils module composer.utils.import_helpers module composer.utils.inference module composer.utils.iter_helpers module composer.utils.misc module composer.utils.module_surgery module composer.utils.object_store module composer.utils.object_store.libcloud_object_store module composer.utils.object_store.object_store module composer.utils.object_store.object_store_hparams module composer.utils.object_store.s3_object_store module composer.utils.object_store.sftp_object_store module composer.utils.reproducibility module composer.utils.retrying module composer.utils.string_enum module composer_deeplabv3() (in module composer.models.deeplabv3.model) composer_efficientnetb0() (in module composer.models.efficientnetb0.model) composer_resnet() (in module composer.models.resnet.model) composer_resnet_cifar() (in module composer.models.resnet_cifar.model) composer_timm() (in module composer.models.timm.model) ComposerClassifier (class in composer.models.tasks.classification) ComposerEnv (class in composer.utils.collect_env) ComposerModel (class in composer.models.base) ComposerScheduler (class in composer.optim.scheduler) compute_ema() (in module composer.algorithms.ema.ema) configure_deterministic_mode() (in module composer.utils.reproducibility) configure_excepthook() (in module composer.utils.collect_env) ConstantScheduler (class in composer.optim.scheduler) ConstantWithWarmupScheduler (class in composer.optim.scheduler) CONTIGUOUS_FORMAT (composer.core.types.MemoryFormat attribute) contrast() (in module composer.algorithms.utils.augmentation_primitives) contrast_original() (in module composer.algorithms.utils.augmentation_primitives) copy() (composer.core.time.Timestamp method) CosineAnnealingScheduler (class in composer.optim.scheduler) CosineAnnealingWarmRestartsScheduler (class in composer.optim.scheduler) CosineAnnealingWithWarmupScheduler (class in composer.optim.scheduler) count_module_instances() (in module composer.utils.module_surgery) count_op_instances() (in module composer.utils.fx_utils) counter() (composer.profiler.marker.Marker method) create_bert_classification() (in module composer.models.bert.model) create_bert_mlm() (in module composer.models.bert.model) create_gpt2() (in module composer.models.gpt2.model) create_symlink_file() (in module composer.utils.file_helpers) current_metrics (composer.core.state.State attribute) CutMix (class in composer.algorithms.cutmix.cutmix) cutmix_batch() (in module composer.algorithms.cutmix.cutmix) CutOut (class in composer.algorithms.cutout.cutout) cutout_batch() (in module composer.algorithms.cutout.cutout) cyclic_schedule() (in module composer.profiler.profiler_schedule) D data (composer.loggers.in_memory_logger.InMemoryLogger attribute) data() (composer.loggers.logger.Logger method) data_batch() (composer.loggers.logger.Logger method) data_epoch() (composer.loggers.logger.Logger method) data_fit() (composer.loggers.logger.Logger method) dataloader (composer.core.state.State property) dataloader_label (composer.core.state.State property) dataloader_len (composer.core.state.State property) DataLoaderHparams (class in composer.datasets.dataset_hparams) Dataset (in module composer.core.types) dataset_registry (in module composer.datasets.dataset_hparams_registry) DatasetHparams (class in composer.datasets.dataset_hparams) DataSpec (class in composer.core.data_spec) ddp_sync_context() (in module composer.trainer.ddp) DDPSyncStrategy (class in composer.trainer.ddp) decode_class() (composer.datasets.cifar.StreamingCIFAR10 method) (composer.datasets.imagenet.StreamingImageNet1k method) decode_image() (composer.datasets.cifar.StreamingCIFAR10 method) (composer.datasets.imagenet.StreamingImageNet1k method) DecoupledAdamW (class in composer.optim.decoupled_weight_decay) DecoupledAdamWHparams (class in composer.optim.optimizer_hparams_registry) DecoupledSGDW (class in composer.optim.decoupled_weight_decay) DecoupledSGDWHparams (class in composer.optim.optimizer_hparams_registry) deeplabv3() (in module composer.models.deeplabv3.model) DeepLabV3Hparams (class in composer.models.deeplabv3.deeplabv3_hparams) deepspeed_enabled (composer.core.state.State property) (composer.trainer.trainer.Trainer property) deepspeed_model (composer.core.state.State property) destinations (composer.loggers.logger.Logger attribute) Device (class in composer.trainer.devices.device) device_num_workers (composer.datasets.streaming.world.World attribute) device_worker (composer.datasets.streaming.world.World attribute) DeviceCPU (class in composer.trainer.devices.device_cpu) DeviceGPU (class in composer.trainer.devices.device_gpu) DeviceMPS (class in composer.trainer.devices.device_mps) DeviceTPU (class in composer.trainer.devices.device_tpu) disable_env_report() (in module composer.utils.collect_env) dispatch_download() (in module composer.datasets.streaming.download) dist_backend (composer.trainer.devices.device.Device attribute) download() (composer.datasets.streaming.dataset.StreamingDataset method) download_checkpoint() (in module composer.utils.checkpoint) download_from_http() (in module composer.datasets.streaming.download) download_from_local() (in module composer.datasets.streaming.download) download_object() (composer.utils.object_store.object_store.ObjectStore method) download_or_wait() (in module composer.datasets.streaming.download) dump() (composer.datasets.streaming.format.StreamingDatasetIndex method) dumps() (composer.datasets.streaming.format.StreamingDatasetIndex method) DURATION (composer.core.time.TimeUnit attribute) E EarlyStopper (class in composer.callbacks.early_stopper) EfficientNet (class in composer.models.efficientnetb0.efficientnets) EfficientNetB0Hparams (class in composer.models.efficientnetb0.efficientnetb0_hparams) EMA (class in composer.algorithms.ema.ema) enable_env_report() (in module composer.utils.collect_env) Encoder (class in composer.models.ssd.utils) Engine (class in composer.core.engine) engine (composer.trainer.trainer.Trainer attribute) ensure_data_spec() (in module composer.core.data_spec) ensure_evaluator() (in module composer.core.evaluator) ensure_folder_has_no_conflicting_files() (in module composer.utils.file_helpers) ensure_folder_is_empty() (in module composer.utils.file_helpers) ensure_time() (in module composer.core.time) ensure_tuple() (in module composer.utils.iter_helpers) epoch (composer.core.time.Timestamp property) EPOCH (composer.core.time.TimeUnit attribute) (composer.loggers.logger.LogLevel attribute) EPOCH_CHECKPOINT (composer.core.event.Event attribute) epoch_checkpoint() (composer.core.callback.Callback method) 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) epoch_wct (composer.core.time.Timestamp property) equalize() (in module composer.algorithms.utils.augmentation_primitives) EVAL (composer.core.types.TrainerMode attribute) 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) eval_timestamp (composer.core.state.State attribute) EvalHparams (class in composer.trainer.trainer_hparams) evaluate_periodically() (in module composer.core.evaluator) Evaluator (class in composer.core.evaluator) EvaluatorHparams (class in composer.datasets.evaluator_hparams) evaluators (composer.core.state.State property) (composer.trainer.trainer.Trainer attribute) Event (class in composer.core.event) event (composer.core.engine.Trace attribute) exit_code (composer.core.engine.Trace attribute) ExperimentHparams (class in composer.trainer.trainer_hparams) ExponentialScheduler (class in composer.optim.scheduler) export_for_inference() (composer.trainer.trainer.Trainer method) (in module composer.utils.inference) export_with_logger() (in module composer.utils.inference) ExportForInferenceCallback (class in composer.callbacks.export_for_inference) ExportFormat (class in composer.utils.inference) 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) factorizing_could_speedup() (in module composer.algorithms.factorize.factorize_modules) file_artifact() (composer.loggers.logger.Logger method) FileLogger (class in composer.loggers.file_logger) filename (composer.loggers.file_logger.FileLogger property) find_unused_parameters (composer.algorithms.layer_freezing.layer_freezing.LayerFreezing property) (composer.core.algorithm.Algorithm property) finish() (composer.datasets.streaming.writer.StreamingDatasetWriter method) (composer.profiler.marker.Marker method) FIT (composer.loggers.logger.LogLevel attribute) fit() (composer.trainer.trainer.Trainer method) FIT_END (composer.core.event.Event attribute) fit_end() (composer.core.callback.Callback method) FIT_START (composer.core.event.Event attribute) fit_start() (composer.core.callback.Callback method) FitHparams (class in composer.trainer.trainer_hparams) FORCED_SYNC (composer.trainer.ddp.DDPSyncStrategy attribute) format_log_data_value() (in module composer.loggers.logger) format_name_with_dist() (in module composer.utils.file_helpers) format_name_with_dist_and_time() (in module composer.utils.file_helpers) forward() (composer.algorithms.gated_linear_units.gated_linear_unit_layers.BERTGatedFFOutput method) (composer.models.base.ComposerModel method) (composer.models.ssd.base_model.Loss method) 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_BertIntermediate() (in module composer.algorithms.gated_linear_units.gated_linear_units) from_BertOutput() (in module composer.algorithms.gated_linear_units.gated_linear_units) from_duration() (composer.core.time.Time class method) from_epoch() (composer.core.time.Time class method) from_LayerNorm() (in module composer.algorithms.fused_layernorm.fused_layernorm) from_sample() (composer.core.time.Time class method) from_timestring() (composer.core.time.Time class method) from_token() (composer.core.time.Time class method) fuse_parallel_linears() (in module composer.utils.fx_utils) FusedLayerNorm (class in composer.algorithms.fused_layernorm.fused_layernorm) G GatedLinearUnits (class in composer.algorithms.gated_linear_units.gated_linear_units) GAUSSIAN (composer.datasets.synthetic.SyntheticDataType attribute) generate_synthetic_tokenizer() (in module composer.datasets.synthetic_lm) get() (composer.core.time.Timestamp method) get_accel_model_name() (in module composer.utils.collect_env) get_composer_env() (in module composer.utils.collect_env) get_composer_version() (in module composer.utils.collect_env) get_compression_scheme_basename() (in module composer.datasets.streaming.format) get_cuda_device_count() (in module composer.utils.collect_env) get_elapsed_duration() (composer.core.state.State method) get_ema_model() (composer.algorithms.ema.ema.EMA method) get_file() (in module composer.utils.file_helpers) get_file_artifact() (composer.loggers.logger_destination.LoggerDestination method) get_free_tcp_port() (in module composer.utils.misc) get_global_rank() (in module composer.utils.dist) get_host_processor_cores() (in module composer.utils.collect_env) get_host_processor_name() (in module composer.utils.collect_env) get_index_basename() (in module composer.datasets.streaming.format) get_initializer() (composer.models.initializers.Initializer method) get_kwargs() (composer.utils.object_store.object_store_hparams.ObjectStoreHparams method) get_local_rank() (in module composer.utils.dist) get_local_world_size() (in module composer.utils.collect_env) (in module composer.utils.dist) get_model_from_name() (composer.models.efficientnetb0.efficientnets.EfficientNet static method) (composer.models.resnet_cifar.resnets.ResNetCIFAR static method) get_model_inputs() (composer.models.huggingface.HuggingFaceModel method) get_node_rank() (in module composer.utils.dist) get_node_world_size() (in module composer.utils.collect_env) get_object_size() (composer.utils.object_store.object_store.ObjectStore method) get_object_store() (in module composer.datasets.streaming.download) get_object_store_cls() (composer.utils.object_store.object_store_hparams.ObjectStoreHparams method) get_partition() (composer.datasets.streaming.format.StreamingDatasetIndex method) get_precision_context() (in module composer.core.precision) get_random_seed() (in module composer.utils.reproducibility) get_rng_state() (in module composer.utils.reproducibility) get_sampler() (in module composer.utils.dist) get_shard_basename() (in module composer.datasets.streaming.format) get_state() (composer.core.time.Timestamp method) get_system_description() (in module composer.callbacks.mlperf) get_timeseries() (composer.loggers.in_memory_logger.InMemoryLogger method) get_torch_env() (in module composer.utils.collect_env) get_uri() (composer.utils.object_store.object_store.ObjectStore method) get_uri_for_artifact() (composer.loggers.object_store_logger.ObjectStoreLogger method) get_world() (in module composer.datasets.streaming.world) get_world_size() (in module composer.utils.dist) GhostBatchNorm (class in composer.algorithms.ghost_batchnorm.ghost_batchnorm) glob_filter() (in module composer.utils.checkpoint) global_device (composer.datasets.streaming.world.World attribute) global_node (composer.datasets.streaming.world.World attribute) global_num_devices (composer.datasets.streaming.world.World attribute) global_num_nodes (composer.datasets.streaming.world.World attribute) global_num_workers (composer.datasets.streaming.world.World attribute) global_worker (composer.datasets.streaming.world.World attribute) GLUEHparams (class in composer.datasets.glue_hparams) GPT2Hparams (class in composer.models.gpt2.gpt2_hparams) grad_accum (composer.core.state.State attribute) GradientClipping (class in composer.algorithms.gradient_clipping.gradient_clipping) GradMonitor (class in composer.callbacks.grad_monitor) H has_file_artifact_destination() (composer.loggers.logger.Logger method) HuggingFaceModel (class in composer.models.huggingface) I image_as_type() (in module composer.algorithms.utils.augmentation_common) ImagenetDatasetHparams (class in composer.datasets.imagenet_hparams) ImageVisualizer (class in composer.callbacks.image_visualizer) import_object() (in module composer.utils.import_helpers) 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.datasets.dataset_hparams.DataLoaderHparams method) (composer.datasets.dataset_hparams.DatasetHparams method) (composer.datasets.evaluator_hparams.EvaluatorHparams method) (composer.models.model_hparams.ModelHparams method) (composer.optim.optimizer_hparams_registry.OptimizerHparams method) (composer.trainer.trainer_hparams.EvalHparams method) (composer.trainer.trainer_hparams.ExperimentHparams method) (composer.trainer.trainer_hparams.FitHparams method) Initializer (class in composer.models.initializers) InMemoryLogger (class in composer.loggers.in_memory_logger) instant() (composer.profiler.marker.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_eval (composer.core.event.Event property) is_initialized() (in module composer.utils.dist) is_model_ddp (composer.core.state.State property) is_model_ddp() (in module composer.utils.misc) is_model_deepspeed() (in module composer.utils.misc) is_notebook() (in module composer.utils.misc) is_predict (composer.core.event.Event property) is_tar() (in module composer.utils.file_helpers) iterate_with_callback() (in module composer.utils.iter_helpers) IteratorFileStream (class in composer.utils.iter_helpers) J JSON (in module composer.core.types) JSONTraceHandler (class in composer.profiler.json_trace_handler) L LabelSmoothing (class in composer.algorithms.label_smoothing.label_smoothing) 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) LibcloudObjectStore (class in composer.utils.object_store.libcloud_object_store) LibcloudObjectStoreHparams (class in composer.utils.object_store.object_store_hparams) LinearScheduler (class in composer.optim.scheduler) LinearWithWarmupScheduler (class in composer.optim.scheduler) LMDatasetHparams (class in composer.datasets.lm_dataset_hparams) load() (composer.datasets.streaming.format.StreamingDatasetIndex class method) load_checkpoint() (in module composer.utils.checkpoint) load_model_state() (composer.core.state.State method) load_rng_state() (in module composer.utils.reproducibility) load_state_dict() (composer.core.serializable.Serializable method) (composer.core.state.State method) loads() (composer.datasets.streaming.format.StreamingDatasetIndex class method) log_data() (composer.loggers.logger_destination.LoggerDestination method) log_file_artifact() (composer.loggers.logger_destination.LoggerDestination method) log_to_console() (composer.loggers.progress_bar_logger.ProgressBarLogger method) Logger (class in composer.loggers.logger) logger (composer.models.base.ComposerModel attribute) (composer.trainer.trainer.Trainer attribute) logger_registry (in module composer.loggers.logger_hparams_registry) LoggerDestination (class in composer.loggers.logger_destination) LogLevel (class in composer.loggers.logger) Loss (class in composer.models.ssd.base_model) loss (composer.core.state.State attribute) loss() (composer.models.base.ComposerModel method) LowRankSolution (class in composer.algorithms.factorize.factorize_core) LRMonitor (class in composer.callbacks.lr_monitor) M make_resnet_bottleneck_stochastic() (in module composer.algorithms.stochastic_depth.stochastic_layers) map_collection() (in module composer.utils.iter_helpers) map_pillow_function() (in module composer.algorithms.utils.augmentation_common) Marker (class in composer.profiler.marker) marker() (composer.profiler.profiler.Profiler method) match() (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_duration (composer.core.state.State property) MAX_SEED (in module composer.utils.reproducibility) MemoryFormat (class in composer.core.types) MemoryMonitor (class in composer.callbacks.memory_monitor) merge_traces() (in module composer.profiler.json_trace_merger) metrics() (composer.models.base.ComposerModel method) MissingConditionalImportError MixUp (class in composer.algorithms.mixup.mixup) mixup_batch() (in module composer.algorithms.mixup.mixup) MLPerfCallback (class in composer.callbacks.mlperf) mnist_model() (in module composer.models.classify_mnist.model) MnistClassifierHparams (class in composer.models.classify_mnist.mnist_hparams) MNISTDatasetHparams (class in composer.datasets.mnist_hparams) Model (class in composer.models.classify_mnist.model) model (composer.core.state.State attribute) ModelHparams (class in composer.models.model_hparams) module composer.algorithms composer.algorithms.algorithm_hparams_registry composer.algorithms.alibi composer.algorithms.alibi.alibi composer.algorithms.alibi.attention_surgery_functions composer.algorithms.alibi.attention_surgery_functions.utils 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.ema composer.algorithms.ema.ema composer.algorithms.factorize composer.algorithms.factorize.factorize composer.algorithms.factorize.factorize_core composer.algorithms.factorize.factorize_modules composer.algorithms.fused_layernorm composer.algorithms.fused_layernorm.fused_layernorm composer.algorithms.gated_linear_units composer.algorithms.gated_linear_units.gated_linear_unit_layers composer.algorithms.gated_linear_units.gated_linear_units composer.algorithms.ghost_batchnorm composer.algorithms.ghost_batchnorm.ghost_batchnorm composer.algorithms.gradient_clipping composer.algorithms.gradient_clipping.gradient_clipping 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.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.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.algorithms.warnings composer.callbacks composer.callbacks.callback_hparams_registry composer.callbacks.checkpoint_saver composer.callbacks.early_stopper composer.callbacks.export_for_inference composer.callbacks.grad_monitor composer.callbacks.image_visualizer composer.callbacks.lr_monitor composer.callbacks.memory_monitor composer.callbacks.mlperf composer.callbacks.speed_monitor composer.callbacks.threshold_stopper composer.core composer.core.algorithm composer.core.callback composer.core.data_spec composer.core.engine composer.core.evaluator composer.core.event composer.core.passes composer.core.precision composer.core.serializable composer.core.state composer.core.time composer.core.types composer.datasets composer.datasets.ade20k composer.datasets.ade20k_hparams composer.datasets.brats composer.datasets.brats_hparams composer.datasets.c4 composer.datasets.c4_hparams composer.datasets.cifar composer.datasets.cifar_hparams composer.datasets.coco composer.datasets.coco_hparams composer.datasets.dataset_hparams composer.datasets.dataset_hparams_registry composer.datasets.evaluator_hparams composer.datasets.ffcv_utils composer.datasets.glue_hparams composer.datasets.imagenet composer.datasets.imagenet_hparams composer.datasets.lm_dataset_hparams composer.datasets.mnist_hparams composer.datasets.streaming composer.datasets.streaming.dataset composer.datasets.streaming.download composer.datasets.streaming.format composer.datasets.streaming.world composer.datasets.streaming.writer composer.datasets.synthetic composer.datasets.synthetic_hparams composer.datasets.synthetic_lm composer.datasets.utils composer.functional composer.loggers composer.loggers.file_logger composer.loggers.in_memory_logger composer.loggers.logger composer.loggers.logger_destination composer.loggers.logger_hparams_registry composer.loggers.object_store_logger composer.loggers.progress_bar_logger composer.loggers.tensorboard_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_hparams composer.models.deeplabv3.model composer.models.efficientnetb0 composer.models.efficientnetb0.efficientnetb0_hparams composer.models.efficientnetb0.efficientnets composer.models.efficientnetb0.model composer.models.gpt2 composer.models.gpt2.gpt2_hparams composer.models.gpt2.model composer.models.huggingface composer.models.initializers composer.models.model_hparams composer.models.resnet composer.models.resnet.model composer.models.resnet.resnet_hparams composer.models.resnet_cifar composer.models.resnet_cifar.model composer.models.resnet_cifar.resnet_cifar_hparams composer.models.resnet_cifar.resnets composer.models.ssd composer.models.ssd.base_model composer.models.ssd.ssd composer.models.ssd.ssd300 composer.models.ssd.ssd_hparams composer.models.ssd.utils composer.models.tasks composer.models.tasks.classification composer.models.timm composer.models.timm.model composer.models.timm.timm_hparams composer.models.unet composer.models.unet.model composer.models.unet.unet composer.models.unet.unet_hparams composer.optim composer.optim.decoupled_weight_decay composer.optim.optimizer_hparams_registry composer.optim.scheduler composer.optim.scheduler_hparams_registry composer.profiler composer.profiler.json_trace_handler composer.profiler.json_trace_merger composer.profiler.marker composer.profiler.profiler composer.profiler.profiler_action composer.profiler.profiler_hparams composer.profiler.profiler_schedule composer.profiler.system_profiler composer.profiler.torch_profiler composer.profiler.trace_handler 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_registry composer.trainer.devices.device_mps composer.trainer.devices.device_tpu composer.trainer.trainer composer.trainer.trainer_hparams composer.utils composer.utils.batch_helpers composer.utils.checkpoint composer.utils.collect_env composer.utils.dist composer.utils.file_helpers composer.utils.fx_utils composer.utils.import_helpers composer.utils.inference composer.utils.iter_helpers composer.utils.misc composer.utils.module_surgery composer.utils.object_store composer.utils.object_store.libcloud_object_store composer.utils.object_store.object_store composer.utils.object_store.object_store_hparams composer.utils.object_store.s3_object_store composer.utils.object_store.sftp_object_store composer.utils.reproducibility composer.utils.retrying composer.utils.string_enum module_to_device() (composer.trainer.devices.device.Device method) monitored_barrier() (in module composer.utils.dist) 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) MultiStepScheduler (class in composer.optim.scheduler) MultiStepWithWarmupScheduler (class in composer.optim.scheduler) N name (composer.core.engine.Trace attribute) node_device (composer.datasets.streaming.world.World attribute) node_num_devices (composer.datasets.streaming.world.World attribute) node_num_workers (composer.datasets.streaming.world.World attribute) node_worker (composer.datasets.streaming.world.World attribute) NoEffectWarning NoOpModel (class in composer.algorithms.no_op_model.no_op_model) NoOpModelClass (class in composer.algorithms.no_op_model.no_op_model) NormalizationFn (class in composer.datasets.utils) NotIntendedUseWarning O object_store (composer.loggers.object_store_logger.ObjectStoreLogger property) ObjectStore (class in composer.utils.object_store.object_store) ObjectStoreHparams (class in composer.utils.object_store.object_store_hparams) ObjectStoreLogger (class in composer.loggers.object_store_logger) ObjectStoreLoggerHparams (class in composer.loggers.logger_hparams_registry) ObjectStoreTransientError onnx (composer.utils.inference.ExportFormat attribute) optimizer_cls (composer.optim.optimizer_hparams_registry.AdamHparams attribute) (composer.optim.optimizer_hparams_registry.AdamWHparams attribute) (composer.optim.optimizer_hparams_registry.DecoupledAdamWHparams attribute) (composer.optim.optimizer_hparams_registry.DecoupledSGDWHparams attribute) (composer.optim.optimizer_hparams_registry.RAdamHparams attribute) (composer.optim.optimizer_hparams_registry.RMSpropHparams attribute) (composer.optim.optimizer_hparams_registry.SGDHparams attribute) optimizer_to_device() (composer.trainer.devices.device.Device method) OptimizerHparams (class in composer.optim.optimizer_hparams_registry) optimizers (composer.core.state.State property) 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) PhotometricDistoration (class in composer.datasets.ade20k) pil_image_collate() (in module composer.datasets.utils) PolicyRegistry (class in composer.algorithms.alibi.attention_surgery_functions.utils) PolynomialScheduler (class in composer.optim.scheduler) PolynomialWithWarmupScheduler (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) PREDICT (composer.core.types.TrainerMode attribute) predict() (composer.trainer.trainer.Trainer method) predict_after_forward() (composer.core.callback.Callback method) predict_batch_end() (composer.core.callback.Callback method) predict_batch_start() (composer.core.callback.Callback method) predict_before_forward() (composer.core.callback.Callback method) predict_end() (composer.core.callback.Callback method) predict_start() (composer.core.callback.Callback method) predict_timestamp (composer.core.state.State attribute) prepare_ddp_module() (in module composer.trainer.ddp) PRESERVE_FORMAT (composer.core.types.MemoryFormat attribute) print_env() (in module composer.utils.collect_env) process_chrome_json_trace_file() (composer.profiler.trace_handler.TraceHandler method) process_counter_event() (composer.profiler.trace_handler.TraceHandler method) process_duration_event() (composer.profiler.trace_handler.TraceHandler method) process_instant_event() (composer.profiler.trace_handler.TraceHandler method) Profiler (class in composer.profiler.profiler) profiler (composer.core.state.State attribute) profiler_scheduler_registry (in module composer.profiler.profiler_hparams) ProfilerAction (class in composer.profiler.profiler_action) ProgressBarLogger (class in composer.loggers.progress_bar_logger) ProgressiveResizing (class in composer.algorithms.progressive_resizing.progressive_resizing) PyTorchScheduler (in module composer.core.types) R RAdamHparams (class in composer.optim.optimizer_hparams_registry) RandAugment (class in composer.algorithms.randaugment.randaugment) randaugment_image() (in module composer.algorithms.randaugment.randaugment) 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) rank_zero_seed (composer.core.state.State attribute) read_array() (in module composer.datasets.streaming.format) record_chrome_json_trace_file() (composer.profiler.profiler.Profiler method) register() (composer.algorithms.alibi.attention_surgery_functions.utils.PolicyRegistry method) register_alibi() (in module composer.algorithms.alibi.attention_surgery_functions.utils) register_pass() (composer.core.engine.Engine method) replace_module_classes() (in module composer.utils.module_surgery) replace_op() (in module composer.utils.fx_utils) ReplacementFunction (in module composer.utils.module_surgery) resize_batch() (in module composer.algorithms.progressive_resizing.progressive_resizing) ResNet9 (class in composer.models.resnet_cifar.resnets) ResNetCIFAR (class in composer.models.resnet_cifar.resnets) ResNetCIFAR.Block (class in composer.models.resnet_cifar.resnets) ResNetCIFARHparams (class in composer.models.resnet_cifar.resnet_cifar_hparams) ResNetHparams (class in composer.models.resnet.resnet_hparams) retry() (in module composer.utils.retrying) RMSpropHparams (class in composer.optim.optimizer_hparams_registry) rotate() (in module composer.algorithms.utils.augmentation_primitives) run (composer.core.engine.Trace attribute) run_event() (composer.core.callback.Callback method) (composer.core.engine.Engine method) run_local_rank_zero_first() (in module composer.utils.dist) run_marker_only_event() (composer.core.engine.Engine method) run_name (composer.core.state.State attribute) S S3ObjectStore (class in composer.utils.object_store.s3_object_store) S3ObjectStoreHparams (class in composer.utils.object_store.object_store_hparams) SAM (class in composer.algorithms.sam.sam) SAMOptimizer (class in composer.algorithms.sam.sam) sample (composer.core.time.Timestamp property) SAMPLE (composer.core.time.TimeUnit attribute) sample_dict_to_bytes() (in module composer.datasets.streaming.format) sample_in_epoch (composer.core.time.Timestamp property) sample_stochastic_forward() (in module composer.algorithms.stochastic_depth.stochastic_layers) save_checkpoint() (composer.trainer.trainer.Trainer method) (in module composer.utils.checkpoint) saved_checkpoints (composer.callbacks.checkpoint_saver.CheckpointSaver attribute) (composer.trainer.trainer.Trainer property) saved_traces (composer.profiler.json_trace_handler.JSONTraceHandler attribute) (composer.profiler.torch_profiler.TorchProfiler attribute) scale_back_batch() (composer.models.ssd.utils.Encoder method) scaler (composer.core.state.State attribute) schedulers (composer.core.state.State property) seed (composer.core.state.State property) 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) SEPARABLE (composer.datasets.synthetic.SyntheticDataType attribute) SeqLengthWarmup (class in composer.algorithms.seq_length_warmup.seq_length_warmup) Serializable (class in composer.core.serializable) serialized_attributes (composer.core.state.State attribute) set_batch_sequence_length() (in module composer.algorithms.seq_length_warmup.seq_length_warmup) set_dataloader() (composer.core.state.State method) set_filo_order() (in module composer.core.passes) SFTPObjectStore (class in composer.utils.object_store.sftp_object_store) SFTPObjectStoreHparams (class in composer.utils.object_store.object_store_hparams) SGDHparams (class in composer.optim.optimizer_hparams_registry) sgdw() (composer.optim.decoupled_weight_decay.DecoupledSGDW static method) ShadowModel (class in composer.algorithms.ema.ema) 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.profiler_action.ProfilerAction attribute) smooth_labels() (in module composer.algorithms.label_smoothing.label_smoothing) solarize() (in module composer.algorithms.utils.augmentation_primitives) sort_fused_layernorm_last() (in module composer.core.passes) sort_selective_backprop_first() (in module composer.core.passes) sort_to_back() (in module composer.core.passes) sort_to_front() (in module composer.core.passes) SpeedMonitor (class in composer.callbacks.speed_monitor) split_compression_suffix() (in module composer.datasets.streaming.format) SqueezeExcite (class in composer.algorithms.squeeze_excite.squeeze_excite) SqueezeExcite2d (class in composer.algorithms.squeeze_excite.squeeze_excite) SqueezeExciteConv2d (class in composer.algorithms.squeeze_excite.squeeze_excite) SSD (class in composer.models.ssd.ssd) SSD300 (class in composer.models.ssd.ssd300) SSDCropping (class in composer.models.ssd.utils) SSDHparams (class in composer.models.ssd.ssd_hparams) SSDTransformer (class in composer.models.ssd.utils) start() (composer.profiler.marker.Marker method) State (class in composer.core.state) state (composer.trainer.trainer.Trainer attribute) state_dict() (composer.core.serializable.Serializable method) step() (composer.optim.decoupled_weight_decay.DecoupledAdamW method) (composer.optim.decoupled_weight_decay.DecoupledSGDW method) StepScheduler (class in composer.optim.scheduler) StochasticDepth (class in composer.algorithms.stochastic_depth.stochastic_depth) StreamingADE20k (class in composer.datasets.ade20k) StreamingADE20kHparams (class in composer.datasets.ade20k_hparams) StreamingC4 (class in composer.datasets.c4) StreamingC4Hparams (class in composer.datasets.c4_hparams) StreamingCIFAR10 (class in composer.datasets.cifar) StreamingCIFAR10Hparams (class in composer.datasets.cifar_hparams) StreamingCOCO (class in composer.datasets.coco) StreamingCOCOHparams (class in composer.datasets.coco_hparams) StreamingDataset (class in composer.datasets.streaming.dataset) StreamingDatasetIndex (class in composer.datasets.streaming.format) StreamingDatasetWriter (class in composer.datasets.streaming.writer) StreamingImageNet1k (class in composer.datasets.imagenet) StreamingImageNet1kHparams (class in composer.datasets.imagenet_hparams) StringEnum (class in composer.utils.string_enum) SWA (class in composer.algorithms.swa.swa) synthetic_hf_dataset_builder() (in module composer.datasets.synthetic_lm) SyntheticBatchPairDataset (class in composer.datasets.synthetic) SyntheticDataLabelType (class in composer.datasets.synthetic) SyntheticDataType (class in composer.datasets.synthetic) SyntheticHparamsMixin (class in composer.datasets.synthetic_hparams) SyntheticPILDataset (class in composer.datasets.synthetic) SyntheticTokenizerParams (class in composer.datasets.synthetic_lm) SystemProfiler (class in composer.profiler.system_profiler) T tensor_to_device() (composer.trainer.devices.device.Device method) TensorboardLogger (class in composer.loggers.tensorboard_logger) ThresholdStopper (class in composer.callbacks.threshold_stopper) Time (class in composer.core.time) Timestamp (class in composer.core.time) timestamp (composer.core.state.State attribute) TimeUnit (class in composer.core.time) TimmHparams (class in composer.models.timm.timm_hparams) to_next_batch() (composer.core.time.Timestamp method) to_next_epoch() (composer.core.time.Timestamp method) to_timestring() (composer.core.time.Time method) token (composer.core.time.Timestamp property) TOKEN (composer.core.time.TimeUnit attribute) token_in_epoch (composer.core.time.Timestamp property) TorchProfiler (class in composer.profiler.torch_profiler) torchscript (composer.utils.inference.ExportFormat attribute) total_wct (composer.core.time.Timestamp property) Trace (class in composer.core.engine) trace_handler_registry (in module composer.profiler.profiler_hparams) trace_handlers (composer.profiler.profiler.Profiler property) TraceHandler (class in composer.profiler.trace_handler) Traces (in module composer.core.engine) TRAIN (composer.core.types.TrainerMode attribute) train_dataloader (composer.core.state.State attribute) Trainer (class in composer.trainer.trainer) TrainerHparams (class in composer.trainer.trainer_hparams) TrainerMode (class in composer.core.types) 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.model) (class in composer.models.unet.unet) UnetHparams (class in composer.models.unet.unet_hparams) unit (composer.core.time.Time property) update_params_in_optimizer() (in module composer.utils.module_surgery) upload_object() (composer.utils.object_store.object_store.ObjectStore method) V validate() (composer.models.base.ComposerModel method) value (composer.core.time.Time property) vit_small_patch16() (in module composer.models) W WandBLogger (class in composer.loggers.wandb_logger) WARMUP (composer.profiler.profiler_action.ProfilerAction attribute) warn_if_multiple_loss_interpolation() (in module composer.core.passes) warning_on_one_line() (in module composer.utils.misc) World (class in composer.datasets.streaming.world) write() (composer.loggers.file_logger.FileLogger method) write_ffcv_dataset() (in module composer.datasets.ffcv_utils) write_sample() (composer.datasets.streaming.writer.StreamingDatasetWriter method) write_samples() (composer.datasets.streaming.writer.StreamingDatasetWriter method) Z zero_and_freeze_expand_position_embeddings() (in module composer.algorithms.alibi.attention_surgery_functions.utils)