place optim state_dict loading inside load_checkpoint
#944
Job | Run time |
---|---|
1m 23s | |
7m 31s | |
7m 34s | |
6m 20s | |
6m 15s | |
3m 25s | |
3m 23s | |
3m 26s | |
3m 29s | |
3m 29s | |
3m 25s | |
3m 28s | |
3m 29s | |
3m 36s | |
3m 30s | |
3m 38s | |
3m 46s | |
3m 41s | |
3m 35s | |
3m 30s | |
3m 31s | |
2m 6s | |
2m 17s | |
1h 29m 47s |