Updates Needed in Datasets

[bebark]

The grad_updates_per_step hyperparameter for SAC changed at some point in the last year or two from being a decimal value to being an integer. I believe the old values are just the inversions of the new (e.g. 8 is 0.125). Is that correct? Either way the brax/datasets readme and data are out of sync with the rest of the repo since they follow the old decimal convention.

[LabChameleon]

I had a look at the old training code and I don't think the number of grad_updates_per_step is the inverted decimal number. Considering

brax/brax/training/sac.py

Line 450 in 4672000

num_updates = int(num_envs * grad_updates_per_step)

the number of steps is num_envs * grad_updates_per_step.