sb3_train.py: Train an RL policy with Stable-Baselines3 on dm_control tasks and periodically save policy checkpoints
sb3_collect.py: Load specific checkpoints and collect fixed-length trajectories into .npz datasets with minimal metadata
Python 3.10+
Required:
dm_control + MuJoCo: pip install dm-control mujoco
Reinforcement learning: pip… See the full description on the dataset page:
https://huggingface.co/datasets/Hang917/DynaTraj.
