LICENSE
README.md
pyproject.toml
setup.py
tests/test_loss_equivalence.py
zoobot/__init__.py
zoobot.egg-info/PKG-INFO
zoobot.egg-info/SOURCES.txt
zoobot.egg-info/dependency_links.txt
zoobot.egg-info/requires.txt
zoobot.egg-info/top_level.txt
zoobot/pytorch/__init__.py
zoobot/pytorch/manchester.py
zoobot/pytorch/estimators/__init__.py
zoobot/pytorch/estimators/cuda_check.py
zoobot/pytorch/estimators/custom_layers.py
zoobot/pytorch/estimators/define_model.py
zoobot/pytorch/estimators/efficientnet_custom.py
zoobot/pytorch/predictions/__init__.py
zoobot/pytorch/predictions/predict_on_catalog.py
zoobot/pytorch/training/__init__.py
zoobot/pytorch/training/finetune.py
zoobot/pytorch/training/losses.py
zoobot/pytorch/training/representations.py
zoobot/pytorch/training/tensorboard_writers.py
zoobot/pytorch/training/train_with_pytorch_lightning.py
zoobot/shared/__init__.py
zoobot/shared/benchmark_datasets.py
zoobot/shared/compress_representations.py
zoobot/shared/label_metadata.py
zoobot/shared/load_predictions.py
zoobot/shared/save_predictions.py
zoobot/shared/schemas.py
zoobot/shared/stats.py
zoobot/tensorflow/__init__.py
zoobot/tensorflow/estimators/__init__.py
zoobot/tensorflow/estimators/alexnet_baseline.py
zoobot/tensorflow/estimators/custom_callbacks.py
zoobot/tensorflow/estimators/custom_layers.py
zoobot/tensorflow/estimators/define_model.py
zoobot/tensorflow/estimators/efficientnet_custom.py
zoobot/tensorflow/estimators/efficientnet_standard.py
zoobot/tensorflow/estimators/small_cnn_baseline.py
zoobot/tensorflow/predictions/__init__.py
zoobot/tensorflow/predictions/predict_on_dataset.py
zoobot/tensorflow/predictions/visualize_dirichlet_predictions.py
zoobot/tensorflow/stats/__init__.py
zoobot/tensorflow/stats/coverage.py
zoobot/tensorflow/stats/dirichlet_stats.py
zoobot/tensorflow/stats/mixture_stats.py
zoobot/tensorflow/training/__init__.py
zoobot/tensorflow/training/custom_metrics.py
zoobot/tensorflow/training/finetune.py
zoobot/tensorflow/training/losses.py
zoobot/tensorflow/training/train_with_keras.py
zoobot/tensorflow/training/training_config.py