CUDA error: out of memory CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1 ...
Recently when I was quantifying my model with your code, every time it runs to reconstruction, the "fp_inp, fp_oup = save_inp_oup_data(fp_model, fp_module, cali_data, store_inp=True, store_oup=True, ...
Results that may be inaccessible to you are currently showing.
Hide inaccessible results
Feedback