move the setting of PYTORCH_CUDA_ALLOC_CONF to the cli rather than train module #5085
Job | Run time |
---|---|
1m 7s | |
6m 52s | |
7m 13s | |
5m 14s | |
5m 48s | |
7m 2s | |
5m 33s | |
28m 42s | |
26m 48s | |
27m 36s | |
2h 1m 55s |
Job | Run time |
---|---|
1m 7s | |
6m 52s | |
7m 13s | |
5m 14s | |
5m 48s | |
7m 2s | |
5m 33s | |
28m 42s | |
26m 48s | |
27m 36s | |
2h 1m 55s |