FastTD3/fast_td3
Younggyo Seo 51c55d4a8a
Support Multi-GPU Training (#22)
- Change in isaaclab_env wrapper to explicitly state GPU for each simulation
- Removing jax cache to support multi-gpu environment launch in MuJoCo Playground
- Removing .train() and .eval() in evaluation and rendering to avoid deadlock in multi-gpu training
- Supporting synchronous normalization for multi-gpu training
2025-07-07 10:24:42 -07:00
..
environments Support Multi-GPU Training (#22) 2025-07-07 10:24:42 -07:00
__init__.py Initial Public Release 2025-05-29 01:49:23 +00:00
fast_td3_deploy.py Support FastTD3 + SimbaV2 (#13) 2025-06-15 12:49:59 -07:00
fast_td3_simbav2.py Support Multi-GPU Training (#22) 2025-07-07 10:24:42 -07:00
fast_td3_utils.py Support Multi-GPU Training (#22) 2025-07-07 10:24:42 -07:00
fast_td3.py Support Multi-GPU Training (#22) 2025-07-07 10:24:42 -07:00
hyperparams.py Support Multi-GPU Training (#22) 2025-07-07 10:24:42 -07:00
train_multigpu.py Support Multi-GPU Training (#22) 2025-07-07 10:24:42 -07:00
train.py Support Multi-GPU Training (#22) 2025-07-07 10:24:42 -07:00
training_notebook.ipynb Bug fix -- MTBench evaluation and missing code (#18) 2025-06-25 09:21:04 -07:00