flax
1a75efc0 - Added an error message to the assert in jax_utils.prefetch_to_device(*).

Commit

5 years ago

Added an error message to the assert in jax_utils.prefetch_to_device(*).

References

#346 - Added an error message to the assert in jax_utils.prefetch_to_device(*)

Author

Dan Graur

Parents

Loading