minor

SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"

# run python -m vagen.server.server host=127.0.0.1 port=5001 in a tmux session first

# run python -m vagen.server.server server.host=127.0.0.1 server.port=5001 in a tmux session first

python -m vagen.env.create_dataset \

    --yaml_path "$SCRIPT_DIR/env_config.yaml" \

    algorithm.high_level_gamma=0.95 \

    data.train_files=data/primitive_skill-vision-debug/train.parquet \

    data.val_files=data/primitive_skill-vision-debug/test.parquet \

    data.train_batch_size=32 \

    data.train_batch_size=64 \

    data.val_batch_size=32 \

    data.max_prompt_length=1024 \

    data.max_response_length=128 \