Unsets profile_duration_ms
in Flax ResNet example to prioritize num_profile_steps
.
#13435
Job | Run time |
---|---|
2s | |
16s | |
32s | |
6s | |
29s | |
5m 19s | |
7m 20s | |
3m 46s | |
6m 33s | |
8m 24s | |
13m 25s | |
7m 3s | |
53m 15s |