File size: 788 Bytes
47fccfe
 
 
 
 
 
 
 
 
 
 
a30a48a
 
 
47fccfe
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
# What to inspect. CLI flags override these.

model: lerobot/smolvla_base              # HuggingFace ID or local path
dataset: lerobot/svla_so101_pickplace    # SO101 pick-and-place dataset
episode: 0
num_frames: 8
device: auto                             # auto | cpu | cuda | mps
output_dir: ./outputs
save_individual: true                   # true to also save each frame separately

# Attention analysis options
method: rollout                         # last-layer | rollout | all-layers
cross_attention: true                   # true to capture action-expert → vision cross-attention (slower)
show_heads: true                        # true to save a per-head attention grid for the first frame
raw_attention: false                    # true to skip positional baseline subtraction