Use torch.inference_mode()
for lower memory usage during calibratio…
#43
Job | Run time |
---|---|
2m 15s | |
1m 55s | |
1m 52s | |
1m 57s | |
7m 59s |
torch.inference_mode()
for lower memory usage during calibratio…
#43
Job | Run time |
---|---|
2m 15s | |
1m 55s | |
1m 52s | |
1m 57s | |
7m 59s |