INFO:openpi.shared.download:Downloading gs://openpi-assets/checkpoints/pi05_droid
100%|████████████████████| 11.6G/11.6G [01:15<00:00, 165MiB/s]
INFO:root:Loading model...
INFO:absl:orbax-checkpoint version: 0.11.13
INFO:absl:Restoring checkpoint from /root/.cache/openpi/openpi-assets
INFO:absl:Finished restoring checkpoint in 7.58 seconds
INFO:root:Loaded norm stats from /root/.cache/openpi/openpi-assets
INFO:root:Creating server (host: 3afa4e5ec440, ip: 172.20.0.2)
INFO:websockets.server:server listening on 0.0.0.0:8000
INFO:websockets.server:connection open
Running policy: 100%|████████████| 20/20 [00:07<00:00, 2.55it/s]
client_infer_ms 390.7 29.6 374.9 377.7 390.6
policy_infer_ms 97.1 0.4 97.0 97.0 97.1
server_infer_ms 155.4 5.5 153.8 156.0 157.1
server_prev_total_ms 392.8 29.5 374.9 377.7 413.0
LOG #2
FIRST
INFERENCE
ON A RENTED
GPU
π0.5 / OPENPI / A40
$ python examples/simple_client/main.py
--env DROID --host localhost
INFO:root:Waiting for server at
ws://localhost:8000...
Running policy: 20/20
policy_infer_ms
97.1
server_infer_ms
155.4
client_infer_ms
390.7
MAC / CLIENT
π0.5
A40 48GB
:8000
RUNPOD / 35 MIN / $0.34
SSH TUNNEL :8000
π0.5 97.1 ms
0
100
200
300
400 ms
ROUND TRIP 390.7 ms
MEAN OF 20