Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/EGalahad/vla-scratch
/ types & classes
Types & classes
74 in github.com/EGalahad/vla-scratch
⨍
Functions
336
◇
Types & classes
74
↓ 16 callers
Class
EvalDatasetCfg
vla_scratch/datasets/config.py:30
↓ 9 callers
Class
EvalDataCfg
vla_scratch/datasets/config.py:43
↓ 6 callers
Class
TrainDatasetCfg
vla_scratch/datasets/config.py:37
↓ 5 callers
Class
TrainDataCfg
vla_scratch/datasets/config.py:66
↓ 4 callers
Class
DontBlindConfig
Config for the BlindVLA LeRobot dataset.
vla_scratch/datasets/dont_blind/config.py:12
↓ 4 callers
Class
PiConfig
vla_scratch/policies/pi/config.py:10
↓ 3 callers
Class
VLMOutputs
vla_scratch/policies/modules/vlm_bridge/base.py:16
↓ 2 callers
Class
AdaptiveModulation
vla_scratch/policies/modules/action_expert/cross_attention_dit.py:59
↓ 2 callers
Class
CoTrainConfig
vla_scratch/datasets/bbox_cotrain/config.py:10
↓ 2 callers
Class
DataConfig
vla_scratch/datasets/config.py:12
↓ 2 callers
Class
FieldNormStats
vla_scratch/transforms/normalization.py:16
↓ 2 callers
Class
Normalize
vla_scratch/transforms/normalization.py:173
↓ 2 callers
Class
ToDataSample
vla_scratch/transforms/common.py:21
↓ 2 callers
Class
ToNumpy
vla_scratch/transforms/common.py:50
↓ 2 callers
Class
ToTorch
vla_scratch/transforms/common.py:38
↓ 2 callers
Class
ZmqPolicyClient
vla_scratch/utils/serving/zmq_policy_client.py:8
↓ 1 callers
Class
ActionChunk
vla_scratch/transforms/data_types.py:24
↓ 1 callers
Class
Attention
vla_scratch/policies/modules/action_expert/cross_attention_dit.py:161
↓ 1 callers
Class
DROIDConfig
vla_scratch/datasets/droid/config.py:8
↓ 1 callers
Class
DataSample
vla_scratch/transforms/data_types.py:28
↓ 1 callers
Class
DeNormalize
vla_scratch/transforms/normalization.py:292
↓ 1 callers
Class
DecoderBlock
vla_scratch/policies/modules/action_expert/cross_attention_dit.py:277
↓ 1 callers
Class
DiTConfig
vla_scratch/policies/modules/action_expert/cross_attention_dit.py:136
↓ 1 callers
Class
DiTModel
vla_scratch/policies/modules/action_expert/cross_attention_dit.py:358
↓ 1 callers
Class
DistributedRankAwareBatchSampler
Form global batches, then split evenly across ranks.
vla_scratch/utils/dataloader.py:7
↓ 1 callers
Class
EvalConfig
scripts/eval_policy.py:43
↓ 1 callers
Class
LazyRMSNorm
vla_scratch/policies/modules/action_expert/cross_attention_dit.py:36
↓ 1 callers
Class
LiberoConfig
vla_scratch/datasets/libero/config.py:8
↓ 1 callers
Class
LiberoConfig
vla_scratch/datasets/libero_global/config.py:99
↓ 1 callers
Class
LiberoIPECConfig
vla_scratch/datasets/libero_global/config.py:11
↓ 1 callers
Class
MLP
vla_scratch/policies/modules/action_expert/cross_attention_dit.py:83
↓ 1 callers
Class
NormStatsConfig
scripts/compute_norm_stats.py:52
↓ 1 callers
Class
Observation
vla_scratch/transforms/data_types.py:14
↓ 1 callers
Class
PaligemmaBridge
vla_scratch/policies/modules/vlm_bridge/paligemma/bridge.py:34
↓ 1 callers
Class
PaligemmaPolicyInput
vla_scratch/policies/modules/vlm_bridge/paligemma/processor.py:15
↓ 1 callers
Class
Qwen3VLBridge
vla_scratch/policies/modules/vlm_bridge/qwen/bridge.py:42
↓ 1 callers
Class
QwenPolicyInput
vla_scratch/policies/modules/vlm_bridge/qwen/processor.py:15
↓ 1 callers
Class
RotaryEmbedding
vla_scratch/policies/modules/action_expert/cross_attention_dit.py:103
↓ 1 callers
Class
ServeConfig
scripts/serve_policy.py:44
↓ 1 callers
Class
ServePolicy
scripts/serve_policy.py:123
↓ 1 callers
Class
SmolVLMBridge
vla_scratch/policies/modules/vlm_bridge/smolvlm/bridge.py:33
↓ 1 callers
Class
SmolVLMPolicyInput
vla_scratch/policies/modules/vlm_bridge/smolvlm/processor.py:15
↓ 1 callers
Class
SuffixInput
vla_scratch/policies/pi/policy.py:45
↓ 1 callers
Class
TrainConfig
scripts/train_policy.py:66
↓ 1 callers
Class
TransformedDataset
vla_scratch/transforms/base.py:16
↓ 1 callers
Class
ZmqPolicyServer
ZMQ ROUTER server that handles multiple concurrent clients. Uses a ROUTER socket to accept connections from multiple clients simultaneously.
vla_scratch/utils/serving/zmq_policy_server.py:13
Class
BasePolicy
Minimal policy interface required by training/eval/serving scripts.
vla_scratch/policies/base.py:14
Class
BboxCotrainEvalConfig
examples/bbox_cotrain/simulation.py:44
Class
CoTrainDataset
Minimal LeRobot dataset wrapper for Bbox Cotrain - Keeps actions as-is (no extra transforms). - Emits processed keys defined in `vla_scr
vla_scratch/datasets/bbox_cotrain/dataset.py:157
Class
DROIDDataset
vla_scratch/datasets/droid/dataset.py:23
Class
DontBlindDataset
Minimal LeRobot dataset wrapper for BlindVLA. - Keeps actions as-is (no extra transforms). - Emits processed keys defined in `vla_scratc
vla_scratch/datasets/dont_blind/lerobot_dataset.py:117
Class
DummyDataset
Single-sample dataset providing the fields expected by ToDataSample.
scripts/generation/dummy_dataset.py:22
Class
EagerEpochIterator
Create a fresh iterator each epoch without prefetching.
vla_scratch/helpers/training.py:621
Class
IPECDataset
vla_scratch/datasets/libero_global/dataset_ipec.py:33
Class
LIBERODataset
vla_scratch/datasets/libero/dataset.py:23
Class
LIBERODataset
vla_scratch/datasets/libero_global/dataset.py:22
Class
LiberoActionToGlobal
Invert network action targets back to dataset action format. Expects `sample["actions"]` to contain per-step targets of shape [K, 10]: [Δpo
vla_scratch/datasets/libero_global/transforms.py:135
Class
LiberoActionToLocal
Convert Libero actions into relative 6D pose deltas.
vla_scratch/datasets/libero_global/transforms.py:97
Class
LiberoEvalConfig
examples/libero/simulation.py:37
Class
LiberoGlobalState
Convert history of Libero states into global 6D pose.
vla_scratch/datasets/libero_global/transforms.py:79
Class
LiberoImages
Stack Libero camera streams into a standard tensor layout.
vla_scratch/datasets/libero_global/transforms.py:175
Class
LiberoStateToLocal
Convert history of Libero states into relative 6D pose deltas.
vla_scratch/datasets/libero_global/transforms.py:45
Class
OnScreenRenderEnv
Enable on-screen rendering while keeping offscreen buffers for images.
examples/libero/simulation.py:233
Class
PaligemmaProcessor
Prepare image + prompt inputs for PaliGemma VLM bridges.
vla_scratch/policies/modules/vlm_bridge/paligemma/processor.py:23
Class
PiPolicy
vla_scratch/policies/pi/policy.py:52
Class
PolicyConfig
vla_scratch/policies/config.py:7
Class
PredictBboxConfig
examples/bbox_cotrain/visualize_bbox.py:28
Class
PrefetchingEpochIterator
Prefetch the next epoch's first batch while the current epoch runs.
vla_scratch/helpers/training.py:552
Class
QwenProcessor
Tokenize prompt using Qwen3-VL chat template and produce policy inputs.
vla_scratch/policies/modules/vlm_bridge/qwen/processor.py:27
Class
SmolVLMProcessor
Tokenize prompt using SmolVLM chat template and produce policy inputs.
vla_scratch/policies/modules/vlm_bridge/smolvlm/processor.py:23
Class
TransformFn
vla_scratch/transforms/base.py:8
Class
VLMBridge
vla_scratch/policies/modules/vlm_bridge/base.py:26
Class
VisualizeAttentionConfig
examples/bbox_cotrain/visualize_attention.py:821
Class
WandbCfg
scripts/train_policy.py:59