Actions: google/flax
Actions
Showing runs from all workflows
5,428 workflow runs
5,428 workflow runs
ValueError: Mapped away dimension of inputs passed to vmap should be sharded the same. Got inconsistent axis specs: None vs batch due to split_rngs being replicated.
Flax - Test
#3744:
Pull request #5189
opened
by
copybara-service
bot
ValueError: Mapped away dimension of inputs passed to vmap should be sharded the same. Got inconsistent axis specs: None vs batch due to split_rngs being replicated.
Google GitHub Admin: Actions Workflow Security Scan
#129:
Pull request #5189
opened
by
copybara-service
bot