Hugging Face's logo

kavinrajkrupsurge
/
lampe-sim-data-openvla

Safetensors
openvla
custom_code
Model card Files
xet
 Community
lampe-sim-data-openvla / training_scripts /FINETUNE_MODIFICATIONS.md
kavinrajkrupsurge's picture
kavinrajkrupsurge
Upload folder using huggingface_hub
85c5c21 verified
preview code
|
raw
history blame contribute delete
1.2 kB

Finetune.py Modifications

This repository includes a modified version of OpenVLA's finetune.py script.

Key Modifications

  1. Dataset Type Selection: Added --dataset_type parameter with options:

    • keyboard_controlled (default)
    • lampe_search
    • combined

  2. LAMP Search Support: Added support for lampe_search_dataset with config options:

    • all
    • samples_001_025
    • samples_026_050

  3. Custom Data Directories: Enhanced support for custom dataset directories via --custom_data_dirs

  4. Dataset Configuration: Automatically sets up custom_mixture_spec and custom_data_dirs based on dataset_type

Usage

See TRAINING_GUIDE.md for detailed usage instructions.

Original Script

The original finetune.py is from the OpenVLA repository: https://github.com/priism-center/openvla

Changes Summary

  • Lines 117-118: Added dataset_type and lampe_search_config parameters
  • Lines 126-137: LAMP Search dataset configuration
  • Lines 139-155: Combined dataset configuration
  • Lines 157-170: Keyboard controlled dataset configuration (default)

All other functionality remains unchanged from the original OpenVLA implementation.