configs/gpu.yaml

# GPU-optimized defaults (RTX 3080 / similar).  CLI flags override these.

model: mthirumalai/so101.pnp.1.policy.s1
dataset: mthirumalai/so101.pnp.1
episode: 0
num_frames: 8
device: cuda
task: null
output_dir: ./outputs/mthirumalai/
save_individual: true

# Attention analysis options
method: rollout
cross_attention: true
show_heads: true
raw_attention: true
attn_threshold: 0.5

# Gradient-based attribution
gradient: both
gradient_device: cuda
gradient_seed: 42
smooth_grad: 20
smooth_grad_sigma: 0.15

# Extended attribution features
gradcam_connector: true
gradcam_vlm_layers: "4,8,12,16"
per_step_cross_attention: true
vision_vs_state: true
per_action_dim: true
language_diff: true

# Data export (for web viewer)
export_data: true

# Model internals report
internals_only: false
with_internals: true
internals_frames: 5
entropy_warn: 0.8
entropy_critical: 0.95
entropy_low: 0.1
redundancy_warn: 0.7
redundancy_critical: 0.9