lerobot/examples/3_train_policy.py