Skip to main content
Ctrl+K

stable-pretraining 0.1.2 documentation

Contents:

  • stable-pretraining
  • Releases
  • Contributing
  • Bibliography

API

  • stable_pretraining.backbone
    • MLP
    • Resnet9
    • ConvMixer
    • from_timm
    • from_torchvision
    • set_embedding_dim
    • TeacherStudentWrapper
    • EvalOnly
    • mae
  • stable_pretraining.data
    • DataModule
    • Collator
    • Dataset
    • FromTorchDataset
    • HFDataset
    • Subset
    • GMM
    • MinariStepsDataset
    • MinariEpisodeDataset
    • swiss_roll
    • generate_perlin_noise_2d
    • perlin_noise_3d
    • Categorical
    • ExponentialMixtureNoiseModel
    • ExponentialNormalNoiseModel
    • RepeatedRandomSampler
    • SupervisedBatchSampler
    • RandomBatchSampler
    • fold_views
    • random_split
    • download
    • bulk_download
    • stable_pretraining.data.transforms
    • stable_pretraining.data.dataset_stats
    • stable_pretraining.data.synthetic_data
  • stable_pretraining.forward
  • stable_pretraining.losses
    • NTXEntLoss
    • NegativeCosineSimilarity
    • VICRegLoss
    • BarlowTwinsLoss
  • stable_pretraining.manager
    • Manager
  • stable_pretraining.module
    • Module
  • stable_pretraining.callbacks
    • OnlineProbe
    • OnlineKNN
    • OnlineWriter
    • RankMe
    • LiDAR
    • EarlyStopping
    • TrainerInfo
    • LoggingCallback
    • ModuleSummary
    • SklearnCheckpoint
    • ImageRetrieval
  • stable_pretraining.optim
    • LARS
    • CosineDecayer
    • LinearWarmup
    • LinearWarmupCosineAnnealing
    • LinearWarmupCyclicAnnealing
    • LinearWarmupThreeStepsAnnealing
  • .rst

set_embedding_dim

Contents

  • set_embedding_dim()
  • Examples using set_embedding_dim:

set_embedding_dim#

stable_pretraining.backbone.set_embedding_dim(module, dim, bias=True, expected_input_shape: tuple | list | None = None, expected_output_shape: tuple | list | None = None)[source]#

Examples using set_embedding_dim:#

Supervised Learning Example

Supervised Learning Example

previous

from_torchvision

next

TeacherStudentWrapper

Contents
  • set_embedding_dim()
  • Examples using set_embedding_dim:

By stable-pretraining team

© Copyright 2024, stable-pretraining team.