Remove usages of orbax_utils.save_args_from_target, as this function does nothing (it used to control a checkpointing behavior that has since been optimized away). #932
Job | Run time |
---|---|
12s | |
18s | |
12s | |
8s | |
53s | |
1s | |
1m 11s | |
1m 10s | |
1s | |
1s | |
53s | |
5m 0s |
Job | Run time |
---|---|
12s | |
18s | |
12s | |
8s | |
53s | |
1s | |
1m 11s | |
1m 10s | |
1s | |
1s | |
53s | |
5m 0s |