All modules for which code is available

• quaterion.dataset.indexing_dataset
• quaterion.dataset.label_cache_dataset
• quaterion.dataset.similarity_data_loader
• quaterion.dataset.similarity_dataset
• quaterion.dataset.similarity_samples
• quaterion.dataset.train_collator
• quaterion.distances
• quaterion.distances.base_distance
• quaterion.distances.cosine
• quaterion.distances.dot_product
• quaterion.distances.euclidean
• quaterion.distances.manhattan
• quaterion.eval.accumulators.accumulator
• quaterion.eval.accumulators.group_accumulator
• quaterion.eval.accumulators.pair_accumulator
• quaterion.eval.attached_metric
• quaterion.eval.base_metric
• quaterion.eval.evaluator
• quaterion.eval.group.group_metric
• quaterion.eval.group.retrieval_r_precision
• quaterion.eval.pair.pair_metric
• quaterion.eval.pair.retrieval_precision
• quaterion.eval.pair.retrieval_reciprocal_rank
• quaterion.eval.samplers.base_sampler
• quaterion.eval.samplers.group_sampler
• quaterion.eval.samplers.pair_sampler
• quaterion.loss.arcface_loss
• quaterion.loss.contrastive_loss
• quaterion.loss.extras.pytorch_metric_learning_wrapper
• quaterion.loss.group_loss
• quaterion.loss.multiple_negatives_ranking_loss
• quaterion.loss.online_contrastive_loss
• quaterion.loss.pairwise_loss
• quaterion.loss.similarity_loss
• quaterion.loss.softmax_loss
• quaterion.loss.triplet_loss
• quaterion.main
• quaterion.train.cache.cache_config
• quaterion.train.cache.cache_encoder
• quaterion.train.cache_mixin
• quaterion.train.callbacks.cleanup_callback
• quaterion.train.callbacks.metrics_callback
• quaterion.train.trainable_model
• quaterion.train.xbm.xbm_buffer
• quaterion.train.xbm.xbm_config
• quaterion.utils.enums
• quaterion.utils.progress_bar
• quaterion.utils.utils