Tom Power
60955d7e88
Fix rollout notebook, update light.directional to light.type.
...
PiperOrigin-RevId: 773646227
Change-Id: I3bb54e548b1a74d7523aaa91be0fc3cb980866f3
2025-06-20 04:43:59 -07:00
Copybara-Service
f1954e98cf
Merge pull request #2589 from Rahul-Lashkari:fix-py-rollout-typos
...
PiperOrigin-RevId: 763686962
Change-Id: I6a9c419200f410f60a1801d74ae05862464ad9a1
2025-05-27 02:00:12 -07:00
Rahul Lashkari
7c8930af96
fix(notebook): corrections in python notebooks
2025-04-21 15:38:50 +05:30
Yuval Tassa
dc98ae978d
Publicize the rollout notebook, small improvements to plotting.
...
PiperOrigin-RevId: 730841667
Change-Id: Id365f21cee078f28cb3fa797701c3c5a409607cd
2025-02-25 05:20:41 -08:00
Levi Burner
cfc59164eb
rollout notebook: fixups to benchmarks, plots, text
...
1. MJX benchmarks use jax.lax.scan and block_until_ready
2. Benchmark parameters updated to match RL use case
3. humanoid benchmarks are warmstarted by 2 seconds, humanoid100 by 4 seconds
4. benchmarks use skip_checks and preallocated threadpools
5. Improved advanced usage plots
6. Benchmark no longer includes model initialization in timed section
7. Update notebook text as needed
2025-02-25 03:25:51 -05:00
Yuval Tassa
a0eb985532
rollout notebook: fix bugs due to cell reordering
...
PiperOrigin-RevId: 729567659
Change-Id: I439da667317ed51370d7c8a9fbfc31d11140752c
2025-02-21 10:05:30 -08:00
Yuval Tassa
f9bb6cb3db
Minor updates to rollout notebook:
...
1. Move benchmarks below advanced usage.
2. Request A100 GPU
3. Minor cosmetic changes.
PiperOrigin-RevId: 729538515
Change-Id: I12709cd95f6865e53fd3932ea0b303ee8efc4028
2025-02-21 08:27:33 -08:00
Copybara-Service
7a2ad8fd97
Merge pull request #2436 from aftersomemath:rollout-notebook-pr
...
PiperOrigin-RevId: 728706864
Change-Id: I8f3a507234c4265564a4fee9df0ab8111ff7dd85
2025-02-19 09:42:08 -08:00
Levi Burner
41a873b016
Notebook showcasing the Python rollout module
2025-02-18 10:57:03 -05:00