scripts/_diag.py

import json, sys
d = sys.argv[1] if len(sys.argv) > 1 else "outputs/svdquant_r32_a0.5_w4a8"
c = json.load(open(f"{d}/quant_config.json"))
print(f"W{c['w_bits']}A{c['a_bits']} rank={c['rank']} alpha={c['alpha']} group={c['w_group']}")
print(f"n_calib={c['n_calib']} calib_dir={c['calib_dir']}")
print(f"weight_recon: {c['diag']}")
print(f"summary: {c['summary']}")
print(f"n_layers={len(c['specs'])}")