Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/TencentARC/Moto
/ types & classes
Types & classes
46 in github.com/TencentARC/Moto
⨍
Functions
210
◇
Types & classes
46
↓ 4 callers
Class
DataPrefetcher
common/data/datasets.py:34
↓ 2 callers
Class
Attention
moto_gpt/src/models/trajectory_gpt2.py:291
↓ 2 callers
Class
LMDBDataset_Mix
common/data/datasets.py:85
↓ 2 callers
Class
RGB_PreProcessor
common/processors/rgb_preprocessors.py:40
↓ 1 callers
Class
Block
moto_gpt/src/models/trajectory_gpt2.py:448
↓ 1 callers
Class
CalvinEnvWrapperRaw
moto_gpt/evaluation/robot_manipulation_benchmarks/calvin/calvin_env_wrapper_raw.py:38
↓ 1 callers
Class
LMDViTEmbeddings
latent_motion_tokenizer/src/models/modeling_lmd_vit.py:13
↓ 1 callers
Class
LMDViTModel
latent_motion_tokenizer/src/models/modeling_lmd_vit.py:87
↓ 1 callers
Class
LatentMotionTokenizer_Trainer
latent_motion_tokenizer/src/trainers/latent_motion_tokenizer_trainer.py:21
↓ 1 callers
Class
LinearWarmup_CosineAnnealing
Construct two LR schedulers. First one is for a linear warmup. Second one is for a cosine annealing.
latent_motion_tokenizer/src/trainers/optimizer.py:39
↓ 1 callers
Class
MFormerEmbeddings
latent_motion_tokenizer/src/models/m_former.py:12
↓ 1 callers
Class
MLP
moto_gpt/src/models/trajectory_gpt2.py:418
↓ 1 callers
Class
MotoGPT_Inference
moto_gpt/evaluation/robot_manipulation_benchmarks/simpler/simpler_env_policy_wrapper.py:23
↓ 1 callers
Class
MotoGPT_PolicyWraper
moto_gpt/src/models/moto_gpt_policy_wraper.py:23
↓ 1 callers
Class
MotoGPT_Trainer
moto_gpt/src/trainers/moto_gpt_trainer.py:19
↓ 1 callers
Class
RandomShiftsAug
common/processors/rgb_preprocessors.py:9
↓ 1 callers
Class
TheiaOutput
latent_motion_tokenizer/src/models/theia_model.py:44
↓ 1 callers
Class
TimmDinoV2Output
latent_motion_tokenizer/src/models/timm_dinov2_model.py:8
↓ 1 callers
Class
TimmDinoV2VsionEncoder
latent_motion_tokenizer/src/models/timm_dinov2_model.py:22
↓ 1 callers
Class
TimmDinoV2VsionEncoder
moto_gpt/src/models/timm_dinov2_model.py:24
↓ 1 callers
Class
ViTPooler
latent_motion_tokenizer/src/models/modeling_lmd_vit.py:72
↓ 1 callers
Class
ViTPooler
latent_motion_tokenizer/src/models/m_former.py:61
Class
ActionEncoding
Defines supported action encoding schemes for different datasets.
data_preprocessing/oxe_dataset_configs.py:29
Class
AdapterMLP
moto_gpt/src/models/trajectory_gpt2.py:433
Class
GPT2Config
This is the configuration class to store the configuration of a :class:`~transformers.GPT2Model` or a :class:`~transformers.TFGPT2Model`. It
moto_gpt/src/models/trajectory_gpt2.py:80
Class
GPT2DoubleHeadsModelOutput
Base class for outputs of models predicting if two sentences are consecutive or not. Args: loss (:obj:`torch.FloatTensor` of shape :o
moto_gpt/src/models/trajectory_gpt2.py:542
Class
GPT2Model
moto_gpt/src/models/trajectory_gpt2.py:687
Class
GPT2PreTrainedModel
An abstract class to handle weights initialization and a simple interface for downloading and loading pretrained models.
moto_gpt/src/models/trajectory_gpt2.py:514
Class
JsonDataset_for_MotoGPT_Video
common/data/datasets.py:327
Class
LMDBDataset_for_MotoGPT
common/data/datasets.py:105
Class
LMDBDataset_for_MotoGPT_CALVIN
common/data/datasets.py:289
Class
LMDBDataset_for_MotoGPT_OXE
common/data/datasets.py:260
Class
LMDBDataset_for_MotoGPT_RT1
common/data/datasets.py:270
Class
LMDBDataset_for_MotoGPT_Video
common/data/datasets.py:265
Class
LatentMotionDecoder
latent_motion_tokenizer/src/models/latent_motion_decoder.py:11
Class
LatentMotionTokenizer
latent_motion_tokenizer/src/models/latent_motion_tokenizer.py:14
Class
MFormer
latent_motion_tokenizer/src/models/m_former.py:75
Class
MaeEncoder
moto_gpt/src/models/mae_model.py:8
Class
MotoGPT
moto_gpt/src/models/moto_gpt.py:24
Class
NpzDataset_for_MotoGPT_Video
common/data/datasets.py:461
Class
ProprioEncoding
Defines supported proprio encoding schemes for different datasets.
data_preprocessing/oxe_dataset_configs.py:18
Class
TheiaEncoder
latent_motion_tokenizer/src/models/theia_model.py:49
Class
TheiaEncoder
moto_gpt/src/models/theia_model.py:49
Class
TheiaOutput
moto_gpt/src/models/theia_model.py:44
Class
TimmDinoV2Output
moto_gpt/src/models/timm_dinov2_model.py:10
Class
VectorQuantizer2
Improved version over VectorQuantizer, can be used as a drop-in replacement. Mostly avoids costly matrix multiplications and allows for pos
latent_motion_tokenizer/src/models/vector_quantizer.py:8