MCPcopy Create free account

hub / github.com/SJTU-DENG-Lab/WLA / types & classes

Types & classes41 in github.com/SJTU-DENG-Lab/WLA

⨍ Functions 234◇ Types & classes 41
↓ 5 callersClassLeRobotEvalDataset
dataset.py:364
↓ 5 callersClassLeRobotTrainDataset
dataset.py:204
↓ 4 callersClassVideoDatasetWrapper
dataset.py:688
↓ 2 callersClassBasicTransformerBlock
models/action_model/cross_attention_dit.py:70
↓ 2 callersClassCategorySpecificLinear
models/action_model/action_model.py:25
↓ 2 callersClassEgodexDatasetWrapper
dataset.py:585
↓ 2 callersClassEpisodeBeginFrameCache
dataset.py:64
↓ 2 callersClassLayerwiseFlowmatchingActionHead
models/action_model/action_model.py:111
↓ 2 callersClassMLP
models/action_model/action_model.py:53
↓ 2 callersClassMultiHeadRMSNorm
models/transformer_encoder.py:20
↓ 2 callersClassSinusoidalPositionalEncoding
Produces a sinusoidal encoding of shape (B, T, w) given timesteps of shape (B, T).
models/action_model/action_encoder.py:24
↓ 2 callersClassVideoRawDataset
dataset.py:638
↓ 1 callersClassActionEncoder
models/action_model/action_model.py:63
↓ 1 callersClassAdaLayerNorm
models/action_model/cross_attention_dit.py:44
↓ 1 callersClassDiT
models/action_model/cross_attention_dit.py:190
↓ 1 callersClassDiffusionModelConfig
train.py:43
↓ 1 callersClassEgodexRawDataset
dataset.py:537
↓ 1 callersClassMLLMInContext
models/model.py:65
↓ 1 callersClassMLLMInContextConfig
models/model.py:20
↓ 1 callersClassPairedDataset
dataset.py:517
↓ 1 callersClassQwen2BidirectionalSdpaAttention
An SDPA-based attention that does NOT apply causal masking. Inherits from Qwen2Attention, but sets self.is_causal = False.
models/transformer_encoder.py:30
↓ 1 callersClassQwen2Encoder
models/transformer_encoder.py:124
↓ 1 callersClassQwen2EncoderLayer
models/transformer_encoder.py:88
↓ 1 callersClassSaveCallback
trainer.py:408
↓ 1 callersClassTimestepEncoder
models/action_model/cross_attention_dit.py:31
↓ 1 callersClassWLA
models/wla.py:58
↓ 1 callersClassWLA0
experiments/libero/wla_utils.py:14
↓ 1 callersClassWLACallback
trainer.py:378
↓ 1 callersClassWLAConfig
models/wla.py:24
↓ 1 callersClassWLATrainer
trainer.py:42
ClassActionEncoder
models/action_model/action_encoder.py:57
ClassCategorySpecificMLP
models/action_model/action_model.py:40
ClassDataArguments
train.py:102
ClassGenerateConfig
experiments/libero/run_libero_eval.py:33
ClassModelArguments
train.py:55
ClassOverrideArguments
train.py:38
ClassProcessorWrapper
trainer_utils.py:9
ClassSelfAttentionTransformer
models/action_model/cross_attention_dit.py:313
ClassTrainingArguments
train.py:120
ClassWLARMBenchPolicy
experiments/rmbench/rmbench_utils.py:158
ClassWLARoboTwinPolicy
experiments/robotwin/robotwin_utils.py:115