Example Syntax Error
guozhaosengzs opened this issue · comments
Small issue, front page of GitHub, on
Augmenting an entire batch of images on the GPU page,
sub_rngs = jax.random.split(rng, images.shape[0])
transformed_images = jax.jit(jax.vmap(transform))(sub_rng, images)
The last sub_rng should be changed to sub_rngs .
Thanks!
Nice catch, thanks for reporting this! :) Fixed with 48e9a99