flax
afaa721e - allow imagenet training profiling to be disabled in config

Commit

1 year ago

allow imagenet training profiling to be disabled in config Set option --config.profile=False on launch to disable. PiperOrigin-RevId: 647849088

References

Author

levskaya

Committer

a-googler

Parents