Skip to main content
Ctrl+K

stable-pretraining 0.1.dev1+gbce7c8b35 documentation

Contents:

  • stable-pretraining
  • Releases
  • Contributing
  • Bibliography

Guides

  • Cache directory & run layout
  • CLI reference

API

  • stable_pretraining.backbone
    • MLP
    • Resnet9
    • ConvMixer
    • from_timm
    • from_torchvision
    • set_embedding_dim
    • TeacherStudentWrapper
    • EvalOnly
  • 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.methods
    • SimCLR
    • BYOL
    • NNCLR
    • MoCov2
    • MoCov3
    • SimSiam
    • PIRL
    • TiCO
    • VICReg
    • VICRegL
    • BarlowTwins
    • WMSE
    • DINO
    • DINOv2
    • DINOv3
    • iBOT
    • SwAV
    • MSN
    • Data2Vec
    • MAE
    • BEiT
    • CMAE
    • MaskFeat
    • SimMIM
    • MIMRefiner
    • iGPT
    • IJEPA
    • LeJEPA
    • SALT
    • NEPA
  • 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
  • stable_pretraining.registry
    • RegistryLogger
    • Registry
    • RunRecord
    • open_registry
  • stable_pretraining.web
    • serve
    • RunScanner
  • .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.