Add Docker Compose evaluation stack (client image + orchestration + docs)#15
Add Docker Compose evaluation stack (client image + orchestration + docs)#15AIWintermuteAI wants to merge 3 commits into
Conversation
|
hi @khanhnd61-vr , |
|
Hi @AIWintermuteAI, yes, we do accept PRs and contributions are very welcome. |
|
Thank you for your reply! |
and DOCKER.md to reflect the changes
|
@khanhnd61-vr okay, merged latest upstream and tested the docker eval workflow again! smolvla-libero eval works great |
Description:
This PR adds a Docker Compose evaluation workflow for running LIBERO /
SimplerEnv simulation episodes against the vla.cpp inference server.
What's included
eval/Dockerfile.client— Python simulation client containerubuntu:22.04with Python 3.10(NumPy 1.x, MuJoCo 2.x, Pandas 2.0.x)
eval/docker-compose.yml— Two-container orchestrationserverservice: builds from the upstreamDockerfile(CUDA)clientservice: Python simulation environmentnvidia-container-toolkitrequired)docs/DOCKER.md— Workflow documentationdocker build/run)Usage
docker compose -f docker-compose.yml build docker compose -f docker-compose.yml up -d server docker compose -f docker-compose.yml run --rm client \ python run_sim_client_direct.py \ --task libero_object --task-id 0 --n-episodes 1 \ --output-dir /tmp/libero_outputs --arch smolvla \ --vla-addr tcp://server:5555Validation
Tested end-to-end on RTX 5060 Ti (CUDA arch 120) with SmolVLA LIBERO: