search

google / flax

Flax is a neural network library for JAX that is designed for flexibility.
https://flax.readthedocs.io
Apache License 2.0
5.8k stars 613 forks source link

Unsets `profile_duration_ms` in Flax ResNet example to prioritize `num_profile_steps`. #4052

Open copybara-service[bot] opened 1 week ago

copybara-service[bot] commented 1 week ago

Unsets profile_duration_ms in Flax ResNet example to prioritize num_profile_steps.