MCPcopy Create free account

hub / github.com/MotrixLab/MotionDiffuse / types & classes

Types & classes40 in github.com/MotrixLab/MotionDiffuse

↓ 6 callersClassSkeleton
text2motion/utils/skeleton.py:4
↓ 5 callersClassStylizationBlock
text2motion/models/transformer.py:60
↓ 3 callersClassDDPMTrainer
text2motion/trainers/ddpm_trainer.py:29
↓ 3 callersClassMotionTransformer
text2motion/models/transformer.py:288
↓ 2 callersClassFFN
text2motion/models/transformer.py:157
↓ 2 callersClassPositionalEncoding
text2motion/datasets/evaluator_models.py:62
↓ 2 callersClassWordVectorizer
text2motion/utils/word_vectorizer.py:46
↓ 1 callersClassDistributedSampler
text2motion/datasets/dataloader.py:16
↓ 1 callersClassEvaluationDataset
text2motion/datasets/evaluator.py:16
↓ 1 callersClassEvaluatorModelWrapper
text2motion/datasets/evaluator.py:376
↓ 1 callersClassGaussianDiffusion
Utilities for training and sampling diffusion models. Ported directly from here, and then adapted over time to further experimentation.
text2motion/models/gaussian_diffusion.py:312
↓ 1 callersClassLinearTemporalCrossAttention
text2motion/models/transformer.py:122
↓ 1 callersClassLinearTemporalDiffusionTransformerDecoderLayer
text2motion/models/transformer.py:173
↓ 1 callersClassLinearTemporalSelfAttention
text2motion/models/transformer.py:89
↓ 1 callersClassLossSecondMomentResampler
text2motion/models/gaussian_diffusion.py:123
↓ 1 callersClassMMGeneratedDataset
text2motion/datasets/evaluator.py:299
↓ 1 callersClassMotionEncoderBiGRUCo
text2motion/datasets/evaluator_models.py:353
↓ 1 callersClassMovementConvEncoder
text2motion/datasets/evaluator_models.py:79
↓ 1 callersClassTemporalCrossAttention
text2motion/models/transformer.py:229
↓ 1 callersClassTemporalDiffusionTransformerDecoderLayer
text2motion/models/transformer.py:264
↓ 1 callersClassTemporalSelfAttention
text2motion/models/transformer.py:196
↓ 1 callersClassText2MotionDataset
Dataset for Text2Motion generation task.
text2motion/datasets/dataset.py:11
↓ 1 callersClassText2MotionDatasetV2
text2motion/datasets/evaluator.py:139
↓ 1 callersClassTextEncoderBiGRUCo
text2motion/datasets/evaluator_models.py:311
↓ 1 callersClassTrainCompOptions
text2motion/options/train_options.py:4
↓ 1 callersClassUniformSampler
text2motion/models/gaussian_diffusion.py:65
ClassAttLayer
text2motion/datasets/evaluator_models.py:232
ClassBaseOptions
text2motion/options/base_options.py:8
ClassContrastiveLoss
Contrastive loss function. Based on: http://yann.lecun.com/exdb/publis/pdf/hadsell-chopra-lecun-06.pdf
text2motion/datasets/evaluator_models.py:11
ClassLossAwareSampler
text2motion/models/gaussian_diffusion.py:74
ClassLossType
text2motion/models/gaussian_diffusion.py:300
ClassModelMeanType
Which type of output the model predicts.
text2motion/models/gaussian_diffusion.py:276
ClassModelVarType
What is used as the model's output variance. The LEARNED_RANGE option has been added to allow the model to predict values between FIXED_
text2motion/models/gaussian_diffusion.py:286
ClassMotionLenEstimatorBiGRU
text2motion/datasets/evaluator_models.py:389
ClassMovementConvDecoder
text2motion/datasets/evaluator_models.py:101
ClassScheduleSampler
A distribution over timesteps in the diffusion process, intended to reduce variance of the objective. By default, samplers perform unbias
text2motion/models/gaussian_diffusion.py:30
ClassTestOptions
text2motion/options/evaluate_options.py:4
ClassTextDecoder
text2motion/datasets/evaluator_models.py:187
ClassTextEncoderBiGRU
text2motion/datasets/evaluator_models.py:267
ClassTextVAEDecoder
text2motion/datasets/evaluator_models.py:123