LICENSE
LICENSE_HEADER
README.md
pyproject.toml
setup.py
composer/__init__.py
composer/algorithms/__init__.py
composer/algorithms/_update_table.py
composer/algorithms/algorithm_hparams.py
composer/algorithms/algorithm_registry.py
composer/algorithms/dummy.py
composer/algorithms/functional.py
composer/algorithms/template.py
composer/algorithms/alibi/__init__.py
composer/algorithms/alibi/alibi.py
composer/algorithms/alibi/gpt2_alibi.py
composer/algorithms/augmix/__init__.py
composer/algorithms/augmix/augmix.py
composer/algorithms/blurpool/__init__.py
composer/algorithms/blurpool/blurpool.py
composer/algorithms/blurpool/blurpool_layers.py
composer/algorithms/channels_last/__init__.py
composer/algorithms/channels_last/channels_last.py
composer/algorithms/colout/__init__.py
composer/algorithms/colout/colout.py
composer/algorithms/curriculum_learning/__init__.py
composer/algorithms/curriculum_learning/curriculum_learning.py
composer/algorithms/cutout/__init__.py
composer/algorithms/cutout/cutout.py
composer/algorithms/ghost_batchnorm/__init__.py
composer/algorithms/ghost_batchnorm/ghost_batchnorm.py
composer/algorithms/label_smoothing/__init__.py
composer/algorithms/label_smoothing/label_smoothing.py
composer/algorithms/layer_freezing/__init__.py
composer/algorithms/layer_freezing/layer_freezing.py
composer/algorithms/mixup/__init__.py
composer/algorithms/mixup/mixup.py
composer/algorithms/no_op_model/__init__.py
composer/algorithms/no_op_model/no_op_model.py
composer/algorithms/progressive_resizing/__init__.py
composer/algorithms/progressive_resizing/progressive_resizing.py
composer/algorithms/randaugment/__init__.py
composer/algorithms/randaugment/randaugment.py
composer/algorithms/sam/__init__.py
composer/algorithms/sam/sam.py
composer/algorithms/scale_schedule/__init__.py
composer/algorithms/scale_schedule/scale_schedule.py
composer/algorithms/selective_backprop/__init__.py
composer/algorithms/selective_backprop/selective_backprop.py
composer/algorithms/squeeze_excite/__init__.py
composer/algorithms/squeeze_excite/squeeze_excite.py
composer/algorithms/stochastic_depth/__init__.py
composer/algorithms/stochastic_depth/sample_stochastic_layers.py
composer/algorithms/stochastic_depth/stochastic_depth.py
composer/algorithms/stochastic_depth/stochastic_layers.py
composer/algorithms/swa/__init__.py
composer/algorithms/swa/hparams.py
composer/algorithms/swa/swa.py
composer/callbacks/__init__.py
composer/callbacks/benchmarker.py
composer/callbacks/callback_hparams.py
composer/callbacks/grad_monitor.py
composer/callbacks/lr_monitor.py
composer/callbacks/profiler.py
composer/callbacks/speed_monitor.py
composer/callbacks/timing_monitor.py
composer/callbacks/torch_profiler.py
composer/core/__init__.py
composer/core/algorithm.py
composer/core/callback.py
composer/core/engine.py
composer/core/event.py
composer/core/precision.py
composer/core/serializable.py
composer/core/state.py
composer/core/surgery.py
composer/core/types.py
composer/core/logging/__init__.py
composer/core/logging/base_backend.py
composer/core/logging/logger.py
composer/datasets/__init__.py
composer/datasets/brats.py
composer/datasets/cifar10.py
composer/datasets/cutout.py
composer/datasets/dataloader.py
composer/datasets/hparams.py
composer/datasets/imagenet.py
composer/datasets/lm_datasets.py
composer/datasets/mnist.py
composer/datasets/synthetic.py
composer/loggers/__init__.py
composer/loggers/file_logger.py
composer/loggers/logger_hparams.py
composer/loggers/registry.py
composer/loggers/tqdm_logger.py
composer/loggers/wandb_logger.py
composer/models/__init__.py
composer/models/_update_table.py
composer/models/base.py
composer/models/efficientnets.py
composer/models/initializers.py
composer/models/loss.py
composer/models/model_hparams.py
composer/models/nlp_metrics.py
composer/models/resnets.py
composer/models/transformer_hparams.py
composer/models/transformer_shared.py
composer/models/classify_mnist/__init__.py
composer/models/classify_mnist/mnist_hparams.py
composer/models/classify_mnist/model.py
composer/models/efficientnetb0/__init__.py
composer/models/efficientnetb0/efficientnetb0_hparams.py
composer/models/efficientnetb0/model.py
composer/models/gpt2/__init__.py
composer/models/gpt2/gpt2_hparams.py
composer/models/gpt2/model.py
composer/models/gpt2/scaling_laws_generator.py
composer/models/resnet101/__init__.py
composer/models/resnet101/model.py
composer/models/resnet101/resnet101_hparams.py
composer/models/resnet18/__init__.py
composer/models/resnet18/model.py
composer/models/resnet18/resnet18_hparams.py
composer/models/resnet50/__init__.py
composer/models/resnet50/model.py
composer/models/resnet50/resnet50_hparams.py
composer/models/resnet56_cifar10/__init__.py
composer/models/resnet56_cifar10/model.py
composer/models/resnet56_cifar10/resnet56_cifar10_hparams.py
composer/models/unet/__init__.py
composer/models/unet/layers.py
composer/models/unet/model.py
composer/models/unet/unet.py
composer/models/unet/unet_hparams.py
composer/optim/__init__.py
composer/optim/decoupled_weight_decay.py
composer/optim/optimizer_hparams.py
composer/optim/pytorch_future.py
composer/optim/scheduler.py
composer/trainer/__init__.py
composer/trainer/checkpoint.py
composer/trainer/ddp.py
composer/trainer/entrypoint.py
composer/trainer/scaler.py
composer/trainer/trainer.py
composer/trainer/trainer_hparams.py
composer/trainer/devices/__init__.py
composer/trainer/devices/device.py
composer/trainer/devices/device_cpu.py
composer/trainer/devices/device_gpu.py
composer/trainer/devices/device_hparams.py
composer/utils/__init__.py
composer/utils/augmentation_primitives.py
composer/utils/data.py
composer/utils/ddp.py
composer/utils/determinism.py
composer/utils/iter_helpers.py
composer/utils/precision.py
composer/utils/string_enum.py
composer/utils/tables.py
composer/utils/tensor.py
composer/yamls/algorithms/alibi.yaml
composer/yamls/algorithms/blurpool.yaml
composer/yamls/algorithms/curriculum_learning.yaml
composer/yamls/algorithms/cutout.yaml
composer/yamls/algorithms/ghost_batchnorm.yaml
composer/yamls/algorithms/label_smoothing.yaml
composer/yamls/algorithms/layer_freezing.yaml
composer/yamls/algorithms/mixup.yaml
composer/yamls/algorithms/progressive_resizing.yaml
composer/yamls/algorithms/randaugment.yaml
composer/yamls/algorithms/scale_schedule.yaml
composer/yamls/algorithms/selective_backprop.yaml
composer/yamls/algorithms/squeeze_excite.yaml
composer/yamls/algorithms/stochastic_depth.yaml
composer/yamls/models/classify_mnist.yaml
composer/yamls/models/classify_mnist_cpu.yaml
composer/yamls/models/efficientnetb0.yaml
composer/yamls/models/gpt2_114m.yaml
composer/yamls/models/gpt2_38m.yaml
composer/yamls/models/gpt2_85m.yaml
composer/yamls/models/resnet101.yaml
composer/yamls/models/resnet101_synthetic.yaml
composer/yamls/models/resnet18.yaml
composer/yamls/models/resnet18_synthetic.yaml
composer/yamls/models/resnet50.yaml
composer/yamls/models/resnet50_synthetic.yaml
composer/yamls/models/resnet56_cifar10.yaml
composer/yamls/models/resnet56_cifar10_synthetic.yaml
mosaicml.egg-info/PKG-INFO
mosaicml.egg-info/SOURCES.txt
mosaicml.egg-info/dependency_links.txt
mosaicml.egg-info/entry_points.txt
mosaicml.egg-info/requires.txt
mosaicml.egg-info/top_level.txt
tests/__init__.py
tests/conftest.py
tests/test_amp.py
tests/test_daily_regressions.py
tests/test_dataset_registry.py
tests/test_efficientnet.py
tests/test_engine.py
tests/test_events.py
tests/test_hparams.py
tests/test_load.py
tests/test_logger.py
tests/test_model_registry.py
tests/test_state.py
tests/test_string_enum.py
tests/test_surgery.py
tests/test_weekly_regressions.py
tests/algorithms/__init__.py
tests/algorithms/algorithm_fixtures.py
tests/algorithms/test_algorithm_registry.py
tests/algorithms/test_blurpool.py
tests/algorithms/test_blurpool_algorithm.py
tests/algorithms/test_channels_last.py
tests/algorithms/test_colout.py
tests/algorithms/test_cutout.py
tests/algorithms/test_ghost_batchnorm.py
tests/algorithms/test_label_smoothing.py
tests/algorithms/test_layer_freezing.py
tests/algorithms/test_mixup.py
tests/algorithms/test_progressive_resizing.py
tests/algorithms/test_scale_schedule.py
tests/algorithms/test_selective_backprop.py
tests/algorithms/test_squeeze_excite.py
tests/algorithms/test_stochastic_depth.py
tests/callbacks/__init__.py
tests/callbacks/test_callbacks.py
tests/callbacks/test_grad_monitor.py
tests/callbacks/test_speed_monitor.py
tests/fixtures/__init__.py
tests/fixtures/ddp_fixtures.py
tests/fixtures/dummy_fixtures.py
tests/trainers/__init__.py
tests/trainers/test_checkpoint.py
tests/trainers/test_ddp.py
tests/trainers/test_optimizer.py
tests/trainers/test_scheduler.py
tests/trainers/test_trainer.py
tests/utils/__init__.py
tests/utils/classifer.py
tests/utils/deep_compare.py
tests/utils/model.py
tests/utils/trainer_fit.py