LICENSE
README.md
pyproject.toml
setup.py
tests/test_contrastive_embedding_collection.py
tests/test_data_augmentation.py
tests/test_dataset.py
tests/test_dataset_online_mapping.py
tests/test_engine.py
tests/test_mae_loss.py
tests/test_matton_augment_torch.py
tests/test_pipeline.py
ume/__init__.py
ume/data/__init__.py
ume/data/dataset.py
ume/data/dataset_online.py
ume/data_augmentation/__init__.py
ume/data_augmentation/matton_augment.py
ume/data_augmentation/matton_augment_torch.py
ume/engine/__init__.py
ume/engine/base.py
ume/engine/contrastive.py
ume/engine/mantis_contrastive.py
ume/engine/reconstruction.py
ume/model/__init__.py
ume/model/efficientnet_contrastive.py
ume/model/encoder_decoder.py
ume/model/mae.py
ume/model/mantis_contrastive.py
ume/model/matton_constrastive.py
ume/model/resnet_contrastive.py
ume/model/vae.py
ume/model/loss/__init__.py
ume/model/loss/contrastive.py
ume/model/loss/reconstruction.py
ume/model/nn/__init__.py
ume/model/nn/cnn_encoder.py
ume/model/nn/linear_net.py
ume/pipeline/__init__.py
ume/pipeline/ume_pipeline.py
ume/utils/__init__.py
ume/utils/dataset_scaling.py
ume/utils/logging.py
ume/utils/sample_scaling.py
ume/utils/splitting.py
ume/utils/weights.py
ume_eda.egg-info/PKG-INFO
ume_eda.egg-info/SOURCES.txt
ume_eda.egg-info/dependency_links.txt
ume_eda.egg-info/requires.txt
ume_eda.egg-info/top_level.txt