The above command effectively runs brax/training/agents/ppo/train.py. I resolved the bug by referring to brax/training/agents/bc/train.py and modifying lines train.py ...