[QUANTIZATION] Save & Load Quantized Relay mod

Hi How can I save and load a quantized and calibrated relay mod? This doesnt work:

mod_file = “module_saved.json”

tvm.relay.save(mod, mod_file) loaded_mod = tvm.relay.load(mod_file)

Thanks!