Index A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | R | S | T | U | V | W A actual_queue_length (stable_pretraining.callbacks.queue.OnlineQueue attribute) (stable_pretraining.callbacks.queue.OnlineQueue property) adapt_resnet_for_lowres() (in module stable_pretraining.utils) (in module stable_pretraining.utils.config) AdditiveGaussian (class in stable_pretraining.data.transforms) all_gather() (in module stable_pretraining.utils) (in module stable_pretraining.utils.distributed) all_reduce() (in module stable_pretraining.utils) (in module stable_pretraining.utils.distributed) alphanum_key() (in module stable_pretraining.utils.log_reader) append() (stable_pretraining.utils.nn_modules.OrderedQueue method) (stable_pretraining.utils.nn_modules.UnsortedQueue method) (stable_pretraining.utils.OrderedQueue method) (stable_pretraining.utils.UnsortedQueue method) apply_center_update() (stable_pretraining.losses.dino.DINOv1Loss method) (stable_pretraining.losses.DINOv1Loss method) apply_manual_optimization_patch() (in module stable_pretraining.utils.lightning_patch) apply_masks() (in module stable_pretraining.data.utils) attn_vectors (stable_pretraining.backbone.probe.MultiHeadAttentiveProbe attribute) AutoLinearClassifier (class in stable_pretraining.backbone.probe) AutoTuneMLP (class in stable_pretraining.backbone.probe) B backward() (stable_pretraining.utils.distributed.FullGatherLayer static method) (stable_pretraining.utils.FullGatherLayer static method) barlow_twins_forward() (in module stable_pretraining.forward), [1] BarlowTwinsLoss (class in stable_pretraining.losses), [1] (class in stable_pretraining.losses.joint_embedding) BatchNorm1dNoBias (class in stable_pretraining.utils) (class in stable_pretraining.utils.nn_modules) broadcast_param_to_list() (in module stable_pretraining.utils) (in module stable_pretraining.utils.inspection_utils) bulk_download() (in module stable_pretraining.data), [1] (in module stable_pretraining.data.download) byol_forward() (in module stable_pretraining.forward), [1] BYOLLoss (class in stable_pretraining.losses) (class in stable_pretraining.losses.joint_embedding) BYPASS_VALUE (stable_pretraining.data.transforms.AdditiveGaussian attribute) C Categorical (class in stable_pretraining.data), [1] (class in stable_pretraining.data.synthetic_data) CenterCrop (class in stable_pretraining.data.transforms) CleanUpCallback (class in stable_pretraining.callbacks.cleanup) clear() (stable_pretraining.static.MetaStatic method) clear_cache() (stable_pretraining.backbone.utils.EfficientMaskedTimmViT method) CLIPLoss (class in stable_pretraining.losses) (class in stable_pretraining.losses.multimodal) CLIPZeroShot (class in stable_pretraining.callbacks.clip_zero_shot) collapse_nested_dict() (in module stable_pretraining.config) Collator (class in stable_pretraining.data), [1] (class in stable_pretraining.data.collate) collect() (stable_pretraining.utils.CSVLogAutoSummarizer method) (stable_pretraining.utils.read_csv_logger.CSVLogAutoSummarizer method) ColorJitter (class in stable_pretraining.data.transforms) column_names (stable_pretraining.data.datasets.FromTorchDataset property) (stable_pretraining.data.datasets.HFDataset property) (stable_pretraining.data.datasets.Subset property) (stable_pretraining.data.FromTorchDataset property) (stable_pretraining.data.HFDataset property) (stable_pretraining.data.MinariEpisodeDataset property) (stable_pretraining.data.MinariStepsDataset property) (stable_pretraining.data.Subset property) (stable_pretraining.data.synthetic_data.MinariEpisodeDataset property) (stable_pretraining.data.synthetic_data.MinariStepsDataset property) Compose (class in stable_pretraining.data.transforms) compute_output_dim() (stable_pretraining.backbone.aggregator.TensorAggregator method) compute_pairwise_distances() (in module stable_pretraining.utils) (in module stable_pretraining.utils.distance_metrics) compute_pairwise_distances_chunked() (in module stable_pretraining.utils) (in module stable_pretraining.utils.distance_metrics) Conditional (class in stable_pretraining.data.transforms) configure_model() (stable_pretraining.callbacks.OnlineProbe method) (stable_pretraining.callbacks.probe.OnlineProbe method) (stable_pretraining.callbacks.utils.TrainableCallback method) (stable_pretraining.OnlineProbe method) configure_optimizers() (stable_pretraining.Module method) (stable_pretraining.module.Module method), [1] ContextTargetsMultiBlockMask (class in stable_pretraining.data.transforms) ControlledTransform (class in stable_pretraining.data.transforms) ConvMixer (class in stable_pretraining.backbone) (class in stable_pretraining.backbone.convmixer) CosineDecayer (class in stable_pretraining.optim) (class in stable_pretraining.optim.lr_scheduler) count_parameters() (in module stable_pretraining.static) CPUOffloadCallback (class in stable_pretraining.callbacks.cpu_offload) create_optimizer() (in module stable_pretraining.optim.utils) create_results_table() (in module stable_pretraining.utils.log_reader) create_scheduler() (in module stable_pretraining.optim.lr_scheduler) create_table() (stable_pretraining.utils.log_reader.TableFormatter static method) cross_entropy_loss() (in module stable_pretraining.losses.dino) CSVLogAutoSummarizer (class in stable_pretraining.utils) (class in stable_pretraining.utils.read_csv_logger) D data (stable_pretraining.callbacks.queue.OnlineQueue attribute) (stable_pretraining.callbacks.queue.OnlineQueue property) (stable_pretraining.static.TIMM_EMBEDDINGS attribute) (stable_pretraining.static.TIMM_PARAMETERS attribute) DataModule (class in stable_pretraining.data), [1] (class in stable_pretraining.data.module) Dataset (class in stable_pretraining.data), [1] (class in stable_pretraining.data.datasets) dataset (stable_pretraining.data.datasets.Subset attribute) (stable_pretraining.data.Subset attribute) default() (in module stable_pretraining.callbacks.factories) detach_tensors() (in module stable_pretraining.utils) (in module stable_pretraining.utils.batch_utils) dict_values() (in module stable_pretraining.utils) (in module stable_pretraining.utils.inspection_utils) DictFormat (class in stable_pretraining.data.module) dino_forward() (in module stable_pretraining.forward) DINOv1Loss (class in stable_pretraining.losses) (class in stable_pretraining.losses.dino) dinov2_forward() (in module stable_pretraining.forward) DINOv2Loss (class in stable_pretraining.losses) (class in stable_pretraining.losses.dino) download() (in module stable_pretraining.data), [1] (in module stable_pretraining.data.download) dump() (stable_pretraining.callbacks.writer.OnlineWriter method) (stable_pretraining.OnlineWriter method) dump_csv_logs() (in module stable_pretraining.cli) E EarlyStopping (class in stable_pretraining) (class in stable_pretraining.callbacks) (class in stable_pretraining.callbacks.utils) EfficientMaskedTimmViT (class in stable_pretraining.backbone.utils) EMA (class in stable_pretraining.utils) (class in stable_pretraining.utils.nn_modules) EmbeddingCache (class in stable_pretraining.callbacks.embedding_cache) EnvironmentDumpCallback (class in stable_pretraining.callbacks.env_info) EpochMilestones (class in stable_pretraining.callbacks.earlystop) escape_labels() (in module stable_pretraining.utils.visualization) EvalOnly (class in stable_pretraining.backbone) (class in stable_pretraining.backbone.utils) execute_from_config() (in module stable_pretraining.utils) (in module stable_pretraining.utils.config) ExponentialMixtureNoiseModel (class in stable_pretraining.data), [1] (class in stable_pretraining.data.synthetic_data) ExponentialNormalNoiseModel (class in stable_pretraining.data), [1] (class in stable_pretraining.data.synthetic_data) F fc (stable_pretraining.backbone.probe.AutoLinearClassifier attribute) (stable_pretraining.backbone.probe.LinearProbe attribute) (stable_pretraining.backbone.probe.MultiHeadAttentiveProbe attribute) FeaturesConcat (class in stable_pretraining.backbone.utils) find_module() (in module stable_pretraining.utils) (in module stable_pretraining.utils.config) find_or_create_queue_callback() (in module stable_pretraining.callbacks.queue) flatten_config() (in module stable_pretraining.utils.log_reader) fold_views() (in module stable_pretraining.data), [1] (in module stable_pretraining.data.utils) format_df_to_latex() (in module stable_pretraining.utils) (in module stable_pretraining.utils.visualization) format_metrics_as_dict() (in module stable_pretraining.callbacks.utils) forward() (stable_pretraining.backbone.aggregator.TensorAggregator method) (stable_pretraining.backbone.ConvMixer method) (stable_pretraining.backbone.convmixer.ConvMixer method) (stable_pretraining.backbone.EvalOnly method) (stable_pretraining.backbone.mae.MaskedAutoencoderViT method) (stable_pretraining.backbone.probe.AutoLinearClassifier method) (stable_pretraining.backbone.probe.AutoTuneMLP method) (stable_pretraining.backbone.probe.LinearProbe method) (stable_pretraining.backbone.probe.MultiHeadAttentiveProbe method) (stable_pretraining.backbone.Resnet9 method) (stable_pretraining.backbone.resnet9.ResidualBlock method) (stable_pretraining.backbone.resnet9.Resnet9 method) (stable_pretraining.backbone.TeacherStudentWrapper method), [1] (stable_pretraining.backbone.utils.EfficientMaskedTimmViT method) (stable_pretraining.backbone.utils.EvalOnly method) (stable_pretraining.backbone.utils.FeaturesConcat method) (stable_pretraining.backbone.utils.ReturnEmbedding method) (stable_pretraining.backbone.utils.TeacherStudentWrapper method) (stable_pretraining.data.transforms.UniformTemporalSubsample method) (stable_pretraining.losses.BarlowTwinsLoss method), [1] (stable_pretraining.losses.BYOLLoss method) (stable_pretraining.losses.CLIPLoss method) (stable_pretraining.losses.dino.DINOv1Loss method) (stable_pretraining.losses.dino.DINOv2Loss method) (stable_pretraining.losses.dino.iBOTPatchLoss method) (stable_pretraining.losses.DINOv1Loss method) (stable_pretraining.losses.DINOv2Loss method) (stable_pretraining.losses.iBOTPatchLoss method) (stable_pretraining.losses.joint_embedding.BarlowTwinsLoss method) (stable_pretraining.losses.joint_embedding.BYOLLoss method) (stable_pretraining.losses.joint_embedding.InfoNCELoss method) (stable_pretraining.losses.joint_embedding.NTXEntLoss method) (stable_pretraining.losses.joint_embedding.SwAVLoss method) (stable_pretraining.losses.joint_embedding.VICRegLoss method) (stable_pretraining.losses.multimodal.CLIPLoss method) (stable_pretraining.losses.NegativeCosineSimilarity method), [1] (stable_pretraining.losses.NTXEntLoss method), [1] (stable_pretraining.losses.utils.NegativeCosineSimilarity method) (stable_pretraining.losses.VICRegLoss method), [1] (stable_pretraining.Module method) (stable_pretraining.module.Module method), [1] (stable_pretraining.TeacherStudentWrapper method) (stable_pretraining.utils.distributed.FullGatherLayer static method) (stable_pretraining.utils.EMA method) (stable_pretraining.utils.FullGatherLayer static method) (stable_pretraining.utils.ImageToVideoEncoder method) (stable_pretraining.utils.L2Norm method) (stable_pretraining.utils.nn_modules.EMA method) (stable_pretraining.utils.nn_modules.ImageToVideoEncoder method) (stable_pretraining.utils.nn_modules.L2Norm method) (stable_pretraining.utils.nn_modules.Normalize method) (stable_pretraining.utils.Normalize method) forward_decoder() (stable_pretraining.backbone.mae.MaskedAutoencoderViT method) forward_encoder() (stable_pretraining.backbone.mae.MaskedAutoencoderViT method) forward_hook_fn() (stable_pretraining.callbacks.embedding_cache.EmbeddingCache method) forward_student() (stable_pretraining.backbone.TeacherStudentWrapper method), [1] (stable_pretraining.backbone.utils.TeacherStudentWrapper method) (stable_pretraining.TeacherStudentWrapper method) forward_teacher() (stable_pretraining.backbone.TeacherStudentWrapper method), [1] (stable_pretraining.backbone.utils.TeacherStudentWrapper method) (stable_pretraining.TeacherStudentWrapper method) from_huggingface() (in module stable_pretraining.backbone.utils) from_timm() (in module stable_pretraining.backbone) (in module stable_pretraining.backbone.utils) from_torchvision() (in module stable_pretraining.backbone) (in module stable_pretraining.backbone.utils) FromTorchDataset (class in stable_pretraining.data), [1] (class in stable_pretraining.data.datasets) FullGatherLayer (class in stable_pretraining.utils) (class in stable_pretraining.utils.distributed) G GaussianBlur (class in stable_pretraining.data.transforms) GDriveUploader (class in stable_pretraining.utils) (class in stable_pretraining.utils.gdrive_utils) generate_dae_samples() (in module stable_pretraining.utils) (in module stable_pretraining.utils.data_generation) generate_dm_samples() (in module stable_pretraining.utils) (in module stable_pretraining.utils.data_generation) generate_perlin_noise_2d() (in module stable_pretraining.data), [1] (in module stable_pretraining.data.synthetic_data) generate_ssl_samples() (in module stable_pretraining.utils) (in module stable_pretraining.utils.data_generation) generate_sup_samples() (in module stable_pretraining.utils) (in module stable_pretraining.utils.data_generation) get() (stable_pretraining.static.MetaStatic method) (stable_pretraining.utils.nn_modules.OrderedQueue method) (stable_pretraining.utils.nn_modules.UnsortedQueue method) (stable_pretraining.utils.OrderedQueue method) (stable_pretraining.utils.UnsortedQueue method) get_1d_sincos_pos_embed_from_grid() (in module stable_pretraining.backbone.mae) get_2d_sincos_pos_embed() (in module stable_pretraining.backbone.mae) get_2d_sincos_pos_embed_from_grid() (in module stable_pretraining.backbone.mae) get_best_variant() (stable_pretraining.backbone.probe.AutoTuneMLP method) get_children_modules() (in module stable_pretraining.backbone.utils) get_data_from_batch_or_outputs() (in module stable_pretraining.utils) (in module stable_pretraining.utils.batch_utils) get_folder_id() (stable_pretraining.utils.gdrive_utils.GDriveUploader method) (stable_pretraining.utils.GDriveUploader method) get_folder_url() (stable_pretraining.utils.gdrive_utils.GDriveUploader method) (stable_pretraining.utils.GDriveUploader method) get_lr() (stable_pretraining.optim.lr_scheduler.LinearWarmupCosineAnnealingLR method) get_name() (stable_pretraining.data.transforms.Transform method) get_num_workers() (in module stable_pretraining.data.utils) get_output_shape() (in module stable_pretraining.backbone.utils) (stable_pretraining.backbone.utils.FeaturesConcat static method) get_params() (stable_pretraining.data.transforms.ColorJitter static method) (stable_pretraining.data.transforms.GaussianBlur static method) (stable_pretraining.data.transforms.RandomCrop static method) (stable_pretraining.data.transforms.RandomResizedCrop static method) (stable_pretraining.data.transforms.RandomRotation static method) get_queue_size() (stable_pretraining.utils.gdrive_utils.GDriveUploader method) (stable_pretraining.utils.GDriveUploader method) get_required_fn_parameters() (in module stable_pretraining.utils) (in module stable_pretraining.utils.inspection_utils) get_unsorted() (stable_pretraining.utils.nn_modules.OrderedQueue method) (stable_pretraining.utils.OrderedQueue method) get_variant() (stable_pretraining.backbone.probe.AutoTuneMLP method) GMM (class in stable_pretraining.data), [1] (class in stable_pretraining.data.synthetic_data) H HFDataset (class in stable_pretraining.data), [1] (class in stable_pretraining.data.datasets) human_size() (in module stable_pretraining.callbacks.cleanup) I iBOTPatchLoss (class in stable_pretraining.losses) (class in stable_pretraining.losses.dino) ImageRetrieval (class in stable_pretraining) (class in stable_pretraining.callbacks) (class in stable_pretraining.callbacks.image_retrieval) ImageToVideoEncoder (class in stable_pretraining.utils) (class in stable_pretraining.utils.nn_modules) imshow_with_grid() (in module stable_pretraining.utils.visualization) indices (stable_pretraining.data.datasets.Subset attribute) (stable_pretraining.data.Subset attribute) InfoNCELoss (class in stable_pretraining.losses.joint_embedding) init_and_sync_wandb() (stable_pretraining.Manager method) (stable_pretraining.manager.Manager method), [1] initialize_weights() (stable_pretraining.backbone.mae.MaskedAutoencoderViT method) instantiate_from_config() (in module stable_pretraining.config) instantiated_data (stable_pretraining.Manager property) (stable_pretraining.manager.Manager property) instantiated_module (stable_pretraining.Manager property) (stable_pretraining.manager.Manager property) is_dist() (in module stable_pretraining.utils.config) is_dist_avail_and_initialized() (in module stable_pretraining.utils) (in module stable_pretraining.utils.distributed) is_saved_with_save_to_disk() (stable_pretraining.data.datasets.HFDataset method) (stable_pretraining.data.HFDataset method) is_writing_epoch() (stable_pretraining.callbacks.writer.OnlineWriter method) (stable_pretraining.OnlineWriter method) items() (stable_pretraining.static.MetaStatic method) K keys() (stable_pretraining.backbone.probe.AutoTuneMLP method) (stable_pretraining.static.MetaStatic method) L L2Norm (class in stable_pretraining.utils) (class in stable_pretraining.utils.nn_modules) Lambda (class in stable_pretraining.data.transforms) LARS (class in stable_pretraining.optim) (class in stable_pretraining.optim.lars) LatentViz (class in stable_pretraining.callbacks.latent_viz) latex_escape() (in module stable_pretraining.utils.visualization) LiDAR (class in stable_pretraining) (class in stable_pretraining.callbacks) (class in stable_pretraining.callbacks.lidar) LinearProbe (class in stable_pretraining.backbone.probe) LinearWarmup (class in stable_pretraining.optim) LinearWarmup() (in module stable_pretraining.optim.lr_scheduler) LinearWarmupCosineAnnealing (class in stable_pretraining.optim) LinearWarmupCosineAnnealing() (in module stable_pretraining.optim.lr_scheduler) LinearWarmupCosineAnnealingLR (class in stable_pretraining.optim.lr_scheduler) LinearWarmupCyclicAnnealing (class in stable_pretraining.optim) LinearWarmupCyclicAnnealing() (in module stable_pretraining.optim.lr_scheduler) LinearWarmupThreeStepsAnnealing (class in stable_pretraining.optim) LinearWarmupThreeStepsAnnealing() (in module stable_pretraining.optim.lr_scheduler) ln (stable_pretraining.backbone.probe.MultiHeadAttentiveProbe attribute) load_best_compressed() (in module stable_pretraining.utils.read_csv_logger) load_hparams_from_ckpt() (in module stable_pretraining.utils) (in module stable_pretraining.utils.config) load_state_dict() (stable_pretraining.callbacks.cpu_offload.CPUOffloadCallback method) (stable_pretraining.callbacks.wd_schedule.WeightDecayUpdater method) (stable_pretraining.data.DataModule method), [1] (stable_pretraining.data.module.DataModule method) (stable_pretraining.utils.nn_modules.OrderedQueue method) (stable_pretraining.utils.OrderedQueue method) LocalLogReader (class in stable_pretraining.utils.log_reader) LoggingCallback (class in stable_pretraining) (class in stable_pretraining.callbacks) (class in stable_pretraining.callbacks.trainer_info) LogReader (class in stable_pretraining.utils.log_reader) M mae (in module stable_pretraining.backbone) mae() (in module stable_pretraining.losses) (in module stable_pretraining.losses.reconstruction) main() (in module stable_pretraining.run) Manager (class in stable_pretraining) (class in stable_pretraining.manager), [1] MaskedAutoencoderViT (class in stable_pretraining.backbone.mae) MetaStatic (class in stable_pretraining.static) METRICS_PATTERNS (stable_pretraining.utils.CSVLogAutoSummarizer attribute) (stable_pretraining.utils.read_csv_logger.CSVLogAutoSummarizer attribute) MinariEpisodeDataset (class in stable_pretraining.data), [1] (class in stable_pretraining.data.synthetic_data) MinariStepsDataset (class in stable_pretraining.data), [1] (class in stable_pretraining.data.synthetic_data) MLP (class in stable_pretraining.backbone) (class in stable_pretraining.backbone.mlp) (class in stable_pretraining.backbone.resnet9) module stable_pretraining stable_pretraining.backbone, [1] stable_pretraining.backbone.aggregator stable_pretraining.backbone.convmixer stable_pretraining.backbone.mae stable_pretraining.backbone.mlp stable_pretraining.backbone.probe stable_pretraining.backbone.resnet9 stable_pretraining.backbone.utils stable_pretraining.callbacks, [1] stable_pretraining.callbacks.checkpoint_sklearn stable_pretraining.callbacks.cleanup stable_pretraining.callbacks.clip_zero_shot stable_pretraining.callbacks.cpu_offload stable_pretraining.callbacks.earlystop stable_pretraining.callbacks.embedding_cache stable_pretraining.callbacks.env_info stable_pretraining.callbacks.factories stable_pretraining.callbacks.image_retrieval stable_pretraining.callbacks.knn stable_pretraining.callbacks.latent_viz stable_pretraining.callbacks.lidar stable_pretraining.callbacks.probe stable_pretraining.callbacks.queue stable_pretraining.callbacks.rankme stable_pretraining.callbacks.teacher_student stable_pretraining.callbacks.trainer_info stable_pretraining.callbacks.utils stable_pretraining.callbacks.wd_schedule stable_pretraining.callbacks.writer stable_pretraining.cli stable_pretraining.config stable_pretraining.data, [1] stable_pretraining.data.collate stable_pretraining.data.dataset_stats, [1] stable_pretraining.data.datasets stable_pretraining.data.download stable_pretraining.data.masking stable_pretraining.data.module stable_pretraining.data.sampler stable_pretraining.data.synthetic_data, [1] stable_pretraining.data.transforms, [1] stable_pretraining.data.utils stable_pretraining.forward, [1] stable_pretraining.losses, [1] stable_pretraining.losses.dino stable_pretraining.losses.joint_embedding stable_pretraining.losses.multimodal stable_pretraining.losses.reconstruction stable_pretraining.losses.utils stable_pretraining.manager, [1] stable_pretraining.module, [1] stable_pretraining.optim, [1] stable_pretraining.optim.lars stable_pretraining.optim.lr_scheduler stable_pretraining.optim.utils stable_pretraining.run stable_pretraining.static stable_pretraining.utils stable_pretraining.utils.batch_utils stable_pretraining.utils.config stable_pretraining.utils.data_generation stable_pretraining.utils.distance_metrics stable_pretraining.utils.distributed stable_pretraining.utils.error_handling stable_pretraining.utils.gdrive_utils stable_pretraining.utils.inspection_utils stable_pretraining.utils.lightning_patch stable_pretraining.utils.log_reader stable_pretraining.utils.nn_modules stable_pretraining.utils.read_csv_logger stable_pretraining.utils.timm_to_hf_hub stable_pretraining.utils.visualization Module (class in stable_pretraining) (class in stable_pretraining.module), [1] module (stable_pretraining.callbacks.utils.TrainableCallback property) ModuleSummary (class in stable_pretraining) (class in stable_pretraining.callbacks) (class in stable_pretraining.callbacks.trainer_info) multi_block_mask() (in module stable_pretraining.data.masking) MultiHeadAttentiveProbe (class in stable_pretraining.backbone.probe) MultiViewTransform (class in stable_pretraining.data.transforms) N NAME (stable_pretraining.callbacks.image_retrieval.ImageRetrieval attribute) name (stable_pretraining.data.transforms.Transform property) NAME (stable_pretraining.ImageRetrieval attribute) named_parameters() (stable_pretraining.Module method) (stable_pretraining.module.Module method), [1] NAMES (stable_pretraining.data.MinariEpisodeDataset attribute) (stable_pretraining.data.MinariStepsDataset attribute) (stable_pretraining.data.synthetic_data.MinariEpisodeDataset attribute) (stable_pretraining.data.synthetic_data.MinariStepsDataset attribute) natural_sort() (in module stable_pretraining.utils.log_reader) NegativeCosineSimilarity (class in stable_pretraining.losses), [1] (class in stable_pretraining.losses.utils) nested_get() (stable_pretraining.data.transforms.Transform method) nested_set() (stable_pretraining.data.transforms.Transform method) nested_step() (stable_pretraining.data.MinariEpisodeDataset method) (stable_pretraining.data.MinariStepsDataset method) (stable_pretraining.data.synthetic_data.MinariEpisodeDataset method) (stable_pretraining.data.synthetic_data.MinariStepsDataset method) nnclr_forward() (in module stable_pretraining.forward), [1] norm (stable_pretraining.backbone.probe.AutoLinearClassifier attribute) (stable_pretraining.backbone.probe.LinearProbe attribute) Normalize (class in stable_pretraining.utils) (class in stable_pretraining.utils.nn_modules) NTXEntLoss (class in stable_pretraining.losses), [1] (class in stable_pretraining.losses.joint_embedding) num_variants() (stable_pretraining.backbone.probe.AutoTuneMLP method) O off_diagonal() (in module stable_pretraining.losses) (in module stable_pretraining.losses.utils) on_after_backward() (stable_pretraining.callbacks.teacher_student.TeacherStudentCallback method) (stable_pretraining.TeacherStudentCallback method) on_before_optimizer_step() (stable_pretraining.callbacks.wd_schedule.WeightDecayUpdater method) on_exception() (stable_pretraining.callbacks.cleanup.CleanUpCallback method) (stable_pretraining.callbacks.cpu_offload.CPUOffloadCallback method) on_fit_end() (stable_pretraining.callbacks.cleanup.CleanUpCallback method) on_fit_start() (stable_pretraining.callbacks.teacher_student.TeacherStudentCallback method) (stable_pretraining.callbacks.wd_schedule.WeightDecayUpdater method) (stable_pretraining.TeacherStudentCallback method) on_load_checkpoint() (stable_pretraining.callbacks.checkpoint_sklearn.SklearnCheckpoint method) (stable_pretraining.callbacks.checkpoint_sklearn.StrictCheckpointCallback method) (stable_pretraining.callbacks.checkpoint_sklearn.WandbCheckpoint method) (stable_pretraining.callbacks.SklearnCheckpoint method) (stable_pretraining.SklearnCheckpoint method) on_predict_batch_end() (stable_pretraining.callbacks.OnlineWriter method) (stable_pretraining.callbacks.writer.OnlineWriter method) (stable_pretraining.OnlineWriter method) on_sanity_check_end() (stable_pretraining.callbacks.OnlineWriter method) (stable_pretraining.callbacks.writer.OnlineWriter method) (stable_pretraining.OnlineWriter method) on_sanity_check_start() (stable_pretraining.callbacks.OnlineWriter method) (stable_pretraining.callbacks.writer.OnlineWriter method) (stable_pretraining.OnlineWriter method) on_save_checkpoint() (stable_pretraining.callbacks.checkpoint_sklearn.SklearnCheckpoint method) (stable_pretraining.callbacks.checkpoint_sklearn.WandbCheckpoint method) (stable_pretraining.callbacks.cpu_offload.CPUOffloadCallback method) (stable_pretraining.callbacks.SklearnCheckpoint method) (stable_pretraining.Module method) (stable_pretraining.module.Module method), [1] (stable_pretraining.SklearnCheckpoint method) on_test_batch_end() (stable_pretraining.callbacks.OnlineWriter method) (stable_pretraining.callbacks.writer.OnlineWriter method) (stable_pretraining.OnlineWriter method) on_test_batch_start() (stable_pretraining.callbacks.embedding_cache.EmbeddingCache method) on_train_batch_end() (stable_pretraining.callbacks.latent_viz.LatentViz method) (stable_pretraining.callbacks.OnlineWriter method) (stable_pretraining.callbacks.queue.OnlineQueue method) (stable_pretraining.callbacks.teacher_student.TeacherStudentCallback method) (stable_pretraining.callbacks.writer.OnlineWriter method) (stable_pretraining.OnlineWriter method) (stable_pretraining.TeacherStudentCallback method) on_train_batch_start() (stable_pretraining.callbacks.embedding_cache.EmbeddingCache method) on_train_end() (stable_pretraining.callbacks.cpu_offload.CPUOffloadCallback method) on_train_epoch_end() (stable_pretraining.callbacks.cleanup.CleanUpCallback method) on_train_start() (stable_pretraining.callbacks.cpu_offload.CPUOffloadCallback method) (stable_pretraining.Module method) (stable_pretraining.module.Module method), [1] on_training_epoch_end() (stable_pretraining.callbacks.earlystop.EpochMilestones method) on_validation_batch_end() (stable_pretraining.callbacks.clip_zero_shot.CLIPZeroShot method) (stable_pretraining.callbacks.knn.OnlineKNN method) (stable_pretraining.callbacks.LiDAR method) (stable_pretraining.callbacks.lidar.LiDAR method) (stable_pretraining.callbacks.OnlineKNN method) (stable_pretraining.callbacks.OnlineWriter method) (stable_pretraining.callbacks.RankMe method) (stable_pretraining.callbacks.rankme.RankMe method) (stable_pretraining.callbacks.writer.OnlineWriter method) (stable_pretraining.LiDAR method) (stable_pretraining.OnlineKNN method) (stable_pretraining.OnlineWriter method) (stable_pretraining.RankMe method) on_validation_batch_start() (stable_pretraining.callbacks.embedding_cache.EmbeddingCache method) on_validation_end() (stable_pretraining.callbacks.LoggingCallback method) (stable_pretraining.callbacks.trainer_info.LoggingCallback method) (stable_pretraining.LoggingCallback method) on_validation_epoch_end() (stable_pretraining.callbacks.earlystop.EpochMilestones method) (stable_pretraining.callbacks.image_retrieval.ImageRetrieval method) (stable_pretraining.callbacks.ImageRetrieval method) (stable_pretraining.callbacks.latent_viz.LatentViz method) (stable_pretraining.callbacks.queue.OnlineQueue method) (stable_pretraining.ImageRetrieval method) on_validation_epoch_start() (stable_pretraining.callbacks.image_retrieval.ImageRetrieval method) (stable_pretraining.callbacks.ImageRetrieval method) (stable_pretraining.callbacks.queue.OnlineQueue method) (stable_pretraining.ImageRetrieval method) OnlineKNN (class in stable_pretraining) (class in stable_pretraining.callbacks) (class in stable_pretraining.callbacks.knn) OnlineProbe (class in stable_pretraining) (class in stable_pretraining.callbacks) (class in stable_pretraining.callbacks.probe) OnlineQueue (class in stable_pretraining.callbacks.queue) OnlineWriter (class in stable_pretraining) (class in stable_pretraining.callbacks) (class in stable_pretraining.callbacks.writer) OrderedQueue (class in stable_pretraining.utils) (class in stable_pretraining.utils.nn_modules) P parameters() (stable_pretraining.Module method) (stable_pretraining.module.Module method), [1] patchify() (stable_pretraining.backbone.mae.MaskedAutoencoderViT method) PatchMasking (class in stable_pretraining.data.transforms) perlin_noise_3d() (in module stable_pretraining.data), [1] (in module stable_pretraining.data.synthetic_data) PILGaussianBlur (class in stable_pretraining.data.transforms) predict() (stable_pretraining.Manager method) (stable_pretraining.manager.Manager method) predict_dataloader() (stable_pretraining.data.DataModule method), [1] (stable_pretraining.data.module.DataModule method) predict_step() (stable_pretraining.Module method) (stable_pretraining.module.Module method), [1] print_config() (in module stable_pretraining.run) print_logger_info() (in module stable_pretraining.manager) print_signal_info() (in module stable_pretraining.manager) process_sample() (stable_pretraining.data.Dataset method) (stable_pretraining.data.datasets.Dataset method) projection_module (stable_pretraining.callbacks.latent_viz.LatentViz property) push_timm_to_hf() (in module stable_pretraining.utils.timm_to_hf_hub) R random_masking() (stable_pretraining.backbone.mae.MaskedAutoencoderViT method) random_seed() (in module stable_pretraining.data.transforms) random_split() (in module stable_pretraining.data), [1] (in module stable_pretraining.data.utils) RandomBatchSampler (class in stable_pretraining.data), [1] (class in stable_pretraining.data.sampler) RandomChannelPermutation (class in stable_pretraining.data.transforms) RandomContiguousTemporalSampler (class in stable_pretraining.data.transforms) RandomCrop (class in stable_pretraining.data.transforms) RandomGrayscale (class in stable_pretraining.data.transforms) RandomHorizontalFlip (class in stable_pretraining.data.transforms) RandomMask (class in stable_pretraining.data.transforms) RandomResizedCrop (class in stable_pretraining.data.transforms) RandomRotation (class in stable_pretraining.data.transforms) RandomSolarize (class in stable_pretraining.data.transforms) RankMe (class in stable_pretraining) (class in stable_pretraining.callbacks) (class in stable_pretraining.callbacks.rankme) read() (stable_pretraining.utils.log_reader.LocalLogReader method) (stable_pretraining.utils.log_reader.LogReader method) (stable_pretraining.utils.log_reader.WandbLogReader method) read_config() (stable_pretraining.utils.log_reader.LocalLogReader method) read_local_logs() (in module stable_pretraining.utils.log_reader) read_local_project() (in module stable_pretraining.utils.log_reader) read_project() (stable_pretraining.utils.log_reader.LocalLogReader method) (stable_pretraining.utils.log_reader.WandbLogReader method) read_wandb_project() (in module stable_pretraining.utils.log_reader) read_wandb_run() (in module stable_pretraining.utils.log_reader) recursive_instantiate() (in module stable_pretraining.config) register_lr_scale_hook() (in module stable_pretraining.backbone.utils) RepeatedRandomSampler (class in stable_pretraining.data), [1] (class in stable_pretraining.data.sampler) replace_module() (in module stable_pretraining.utils) (in module stable_pretraining.utils.config) rescale_loss_for_grad_acc() (stable_pretraining.Module method) (stable_pretraining.module.Module method) ResidualBlock (class in stable_pretraining.backbone.resnet9) Resize (class in stable_pretraining.data.transforms) Resnet9 (class in stable_pretraining.backbone) (class in stable_pretraining.backbone.resnet9) restore_original_validation() (in module stable_pretraining.utils.lightning_patch) ReturnEmbedding (class in stable_pretraining.backbone.utils) RGB (class in stable_pretraining.data.transforms) rgetattr() (in module stable_pretraining.utils) (in module stable_pretraining.utils.config) RoundRobinMultiViewTransform (class in stable_pretraining.data.transforms) RoutingTransform (class in stable_pretraining.data.transforms) rsetattr() (in module stable_pretraining.utils) (in module stable_pretraining.utils.config) run() (in module stable_pretraining.cli) S sample() (stable_pretraining.data.Categorical method) (stable_pretraining.data.ExponentialMixtureNoiseModel method) (stable_pretraining.data.ExponentialNormalNoiseModel method) (stable_pretraining.data.synthetic_data.Categorical method) (stable_pretraining.data.synthetic_data.ExponentialMixtureNoiseModel method) (stable_pretraining.data.synthetic_data.ExponentialNormalNoiseModel method) save_best_compressed() (in module stable_pretraining.utils.read_csv_logger) save_checkpoint() (stable_pretraining.Manager method) (stable_pretraining.manager.Manager method) score() (stable_pretraining.data.GMM method) (stable_pretraining.data.synthetic_data.GMM method) set_embedding_dim() (in module stable_pretraining.backbone) (in module stable_pretraining.backbone.utils) set_pl_trainer() (stable_pretraining.data.DataModule method) (stable_pretraining.data.Dataset method) (stable_pretraining.data.datasets.Dataset method) (stable_pretraining.data.MinariEpisodeDataset method) (stable_pretraining.data.module.DataModule method) (stable_pretraining.data.synthetic_data.MinariEpisodeDataset method) set_seed() (in module stable_pretraining.data.transforms) setup() (stable_pretraining.callbacks.checkpoint_sklearn.SklearnCheckpoint method) (stable_pretraining.callbacks.checkpoint_sklearn.WandbCheckpoint method) (stable_pretraining.callbacks.clip_zero_shot.CLIPZeroShot method) (stable_pretraining.callbacks.cpu_offload.CPUOffloadCallback method) (stable_pretraining.callbacks.embedding_cache.EmbeddingCache method) (stable_pretraining.callbacks.env_info.EnvironmentDumpCallback method) (stable_pretraining.callbacks.knn.OnlineKNN method) (stable_pretraining.callbacks.latent_viz.LatentViz method) (stable_pretraining.callbacks.LiDAR method) (stable_pretraining.callbacks.lidar.LiDAR method) (stable_pretraining.callbacks.ModuleSummary method) (stable_pretraining.callbacks.OnlineKNN method) (stable_pretraining.callbacks.queue.OnlineQueue method) (stable_pretraining.callbacks.RankMe method) (stable_pretraining.callbacks.rankme.RankMe method) (stable_pretraining.callbacks.SklearnCheckpoint method) (stable_pretraining.callbacks.trainer_info.ModuleSummary method) (stable_pretraining.callbacks.trainer_info.SLURMInfo method) (stable_pretraining.callbacks.trainer_info.TrainerInfo method) (stable_pretraining.callbacks.TrainerInfo method) (stable_pretraining.data.DataModule method), [1] (stable_pretraining.data.module.DataModule method) (stable_pretraining.LiDAR method) (stable_pretraining.ModuleSummary method) (stable_pretraining.OnlineKNN method) (stable_pretraining.RankMe method) (stable_pretraining.SklearnCheckpoint method) (stable_pretraining.TrainerInfo method) setup_optimizer() (stable_pretraining.callbacks.latent_viz.LatentViz method) (stable_pretraining.callbacks.utils.TrainableCallback method) setup_scheduler() (stable_pretraining.callbacks.utils.TrainableCallback method) should_stop() (stable_pretraining.callbacks.utils.EarlyStopping method) (stable_pretraining.EarlyStopping method) simclr_forward() (in module stable_pretraining.forward), [1] single_nested_get() (stable_pretraining.data.transforms.Transform method) single_nested_set() (stable_pretraining.data.transforms.Transform method) sinkhorn() (stable_pretraining.losses.joint_embedding.SwAVLoss method) sinkhorn_knopp() (in module stable_pretraining.losses) (in module stable_pretraining.losses.utils) sinkhorn_knopp_teacher() (stable_pretraining.losses.dino.DINOv1Loss method) (stable_pretraining.losses.dino.iBOTPatchLoss method) (stable_pretraining.losses.DINOv1Loss method) (stable_pretraining.losses.iBOTPatchLoss method) SklearnCheckpoint (class in stable_pretraining) (class in stable_pretraining.callbacks) (class in stable_pretraining.callbacks.checkpoint_sklearn) SLURMInfo (class in stable_pretraining.callbacks.trainer_info) softmax_center_teacher() (stable_pretraining.losses.dino.DINOv1Loss method) (stable_pretraining.losses.DINOv1Loss method) stable_pretraining module stable_pretraining.backbone module, [1] stable_pretraining.backbone.aggregator module stable_pretraining.backbone.convmixer module stable_pretraining.backbone.mae module stable_pretraining.backbone.mlp module stable_pretraining.backbone.probe module stable_pretraining.backbone.resnet9 module stable_pretraining.backbone.utils module stable_pretraining.callbacks module, [1] stable_pretraining.callbacks.checkpoint_sklearn module stable_pretraining.callbacks.cleanup module stable_pretraining.callbacks.clip_zero_shot module stable_pretraining.callbacks.cpu_offload module stable_pretraining.callbacks.earlystop module stable_pretraining.callbacks.embedding_cache module stable_pretraining.callbacks.env_info module stable_pretraining.callbacks.factories module stable_pretraining.callbacks.image_retrieval module stable_pretraining.callbacks.knn module stable_pretraining.callbacks.latent_viz module stable_pretraining.callbacks.lidar module stable_pretraining.callbacks.probe module stable_pretraining.callbacks.queue module stable_pretraining.callbacks.rankme module stable_pretraining.callbacks.teacher_student module stable_pretraining.callbacks.trainer_info module stable_pretraining.callbacks.utils module stable_pretraining.callbacks.wd_schedule module stable_pretraining.callbacks.writer module stable_pretraining.cli module stable_pretraining.config module stable_pretraining.data module, [1] stable_pretraining.data.collate module stable_pretraining.data.dataset_stats module, [1] stable_pretraining.data.datasets module stable_pretraining.data.download module stable_pretraining.data.masking module stable_pretraining.data.module module stable_pretraining.data.sampler module stable_pretraining.data.synthetic_data module, [1] stable_pretraining.data.transforms module, [1] stable_pretraining.data.utils module stable_pretraining.forward module, [1] stable_pretraining.losses module, [1] stable_pretraining.losses.dino module stable_pretraining.losses.joint_embedding module stable_pretraining.losses.multimodal module stable_pretraining.losses.reconstruction module stable_pretraining.losses.utils module stable_pretraining.manager module, [1] stable_pretraining.module module, [1] stable_pretraining.optim module, [1] stable_pretraining.optim.lars module stable_pretraining.optim.lr_scheduler module stable_pretraining.optim.utils module stable_pretraining.run module stable_pretraining.static module stable_pretraining.utils module stable_pretraining.utils.batch_utils module stable_pretraining.utils.config module stable_pretraining.utils.data_generation module stable_pretraining.utils.distance_metrics module stable_pretraining.utils.distributed module stable_pretraining.utils.error_handling module stable_pretraining.utils.gdrive_utils module stable_pretraining.utils.inspection_utils module stable_pretraining.utils.lightning_patch module stable_pretraining.utils.log_reader module stable_pretraining.utils.nn_modules module stable_pretraining.utils.read_csv_logger module stable_pretraining.utils.timm_to_hf_hub module stable_pretraining.utils.visualization module state_dict() (stable_pretraining.callbacks.cpu_offload.CPUOffloadCallback method) (stable_pretraining.callbacks.wd_schedule.WeightDecayUpdater method) (stable_pretraining.data.DataModule method), [1] (stable_pretraining.data.module.DataModule method) state_key (stable_pretraining.callbacks.image_retrieval.ImageRetrieval property) (stable_pretraining.callbacks.ImageRetrieval property) (stable_pretraining.callbacks.knn.OnlineKNN property) (stable_pretraining.callbacks.LiDAR property) (stable_pretraining.callbacks.lidar.LiDAR property) (stable_pretraining.callbacks.OnlineKNN property) (stable_pretraining.callbacks.RankMe property) (stable_pretraining.callbacks.rankme.RankMe property) (stable_pretraining.callbacks.utils.TrainableCallback property) (stable_pretraining.ImageRetrieval property) (stable_pretraining.LiDAR property) (stable_pretraining.OnlineKNN property) (stable_pretraining.RankMe property) step() (stable_pretraining.optim.LARS method) (stable_pretraining.optim.lars.LARS method) StrictCheckpointCallback (class in stable_pretraining.callbacks.checkpoint_sklearn) Subset (class in stable_pretraining.data), [1] (class in stable_pretraining.data.datasets) supervised_forward() (in module stable_pretraining.forward), [1] SupervisedBatchSampler (class in stable_pretraining.data), [1] (class in stable_pretraining.data.sampler) swapped_prediction() (stable_pretraining.losses.joint_embedding.SwAVLoss method) swav_forward() (in module stable_pretraining.forward) SwAVLoss (class in stable_pretraining.losses.joint_embedding) swiss_roll() (in module stable_pretraining.data), [1] (in module stable_pretraining.data.synthetic_data) T TableFormatter (class in stable_pretraining.utils.log_reader) tabulate_runs() (stable_pretraining.utils.log_reader.TableFormatter static method) TeacherStudentCallback (class in stable_pretraining) (class in stable_pretraining.callbacks.teacher_student) TeacherStudentWrapper (class in stable_pretraining) (class in stable_pretraining.backbone), [1] (class in stable_pretraining.backbone.utils) teardown() (stable_pretraining.callbacks.cpu_offload.CPUOffloadCallback method) (stable_pretraining.callbacks.embedding_cache.EmbeddingCache method) (stable_pretraining.callbacks.env_info.EnvironmentDumpCallback method) (stable_pretraining.data.DataModule method), [1] (stable_pretraining.data.module.DataModule method) TensorAggregator (class in stable_pretraining.backbone.aggregator) test() (stable_pretraining.Manager method) (stable_pretraining.manager.Manager method) test_dataloader() (stable_pretraining.data.DataModule method), [1] (stable_pretraining.data.module.DataModule method) test_step() (stable_pretraining.Module method) (stable_pretraining.module.Module method), [1] TIMM_EMBEDDINGS (class in stable_pretraining.static) TIMM_PARAMETERS (class in stable_pretraining.static) to_image() (in module stable_pretraining.data.transforms) to_scalar() (in module stable_pretraining.callbacks.earlystop) ToImage (class in stable_pretraining.data.transforms) train() (stable_pretraining.backbone.EvalOnly method) (stable_pretraining.backbone.utils.EvalOnly method) train_dataloader() (stable_pretraining.data.DataModule method), [1] (stable_pretraining.data.module.DataModule method) TrainableCallback (class in stable_pretraining.callbacks.utils) TrainerInfo (class in stable_pretraining) (class in stable_pretraining.callbacks) (class in stable_pretraining.callbacks.trainer_info) training_step() (stable_pretraining.Module method) (stable_pretraining.module.Module method), [1] Transform (class in stable_pretraining.data.transforms) U UniformTemporalSubsample (class in stable_pretraining.data.transforms) unpatchify() (stable_pretraining.backbone.mae.MaskedAutoencoderViT method) UnsortedQueue (class in stable_pretraining.utils) (class in stable_pretraining.utils.nn_modules) update() (stable_pretraining.static.MetaStatic method) update_center() (stable_pretraining.losses.dino.DINOv1Loss method) (stable_pretraining.losses.DINOv1Loss method) update_ema_coefficient() (stable_pretraining.backbone.TeacherStudentWrapper method), [1] (stable_pretraining.backbone.utils.TeacherStudentWrapper method) (stable_pretraining.TeacherStudentWrapper method) update_teacher() (stable_pretraining.backbone.TeacherStudentWrapper method), [1] (stable_pretraining.backbone.utils.TeacherStudentWrapper method) (stable_pretraining.TeacherStudentWrapper method) upload_file() (stable_pretraining.utils.gdrive_utils.GDriveUploader method) (stable_pretraining.utils.GDriveUploader method) V val_dataloader() (stable_pretraining.data.DataModule method), [1] (stable_pretraining.data.module.DataModule method) validate() (stable_pretraining.Manager method) (stable_pretraining.manager.Manager method) validation_step() (stable_pretraining.Module method) (stable_pretraining.module.Module method), [1] values() (stable_pretraining.static.MetaStatic method) vicreg_forward() (in module stable_pretraining.forward), [1] VICRegLoss (class in stable_pretraining.losses), [1] (class in stable_pretraining.losses.joint_embedding) visualize_images_graph() (in module stable_pretraining.utils.visualization) vit_base_patch16() (in module stable_pretraining.backbone.mae) vit_base_patch16_dec512d8b() (in module stable_pretraining.backbone.mae) vit_hf() (in module stable_pretraining.backbone.utils) vit_huge_patch14() (in module stable_pretraining.backbone.mae) vit_huge_patch14_dec512d8b() (in module stable_pretraining.backbone.mae) vit_large_patch16() (in module stable_pretraining.backbone.mae) vit_large_patch16_dec512d8b() (in module stable_pretraining.backbone.mae) W wait_for_uploads() (stable_pretraining.utils.gdrive_utils.GDriveUploader method) (stable_pretraining.utils.GDriveUploader method) WandbCheckpoint (class in stable_pretraining.callbacks.checkpoint_sklearn) WandbLogReader (class in stable_pretraining.utils.log_reader) WeightDecayUpdater (class in stable_pretraining.callbacks.wd_schedule) with_hf_retry_ratelimit() (in module stable_pretraining.utils) (in module stable_pretraining.utils.error_handling) wrap_configure_model() (stable_pretraining.callbacks.utils.TrainableCallback method) wrap_configure_optimizers() (stable_pretraining.callbacks.utils.TrainableCallback method) wrap_forward() (stable_pretraining.callbacks.probe.OnlineProbe method) (stable_pretraining.OnlineProbe method) wrap_validation_step() (in module stable_pretraining.callbacks.image_retrieval) WrapTorchTransform (class in stable_pretraining.data.transforms) write_at_phase() (stable_pretraining.callbacks.writer.OnlineWriter method) (stable_pretraining.OnlineWriter method)