Commit c0fd4e00 by zhengzifu

Update Python version to 3.12 and optimize matrix handling in…

Update Python version to 3.12 and optimize matrix handling in generate_wt_group.py by transposing the matrix before shape extraction.
parent 05df0620
......@@ -113,7 +113,8 @@ def run(name: str, config: CFG):
with open(weights_file, "rb") as f:
print(f"Loading {weights_file_name}")
matrixs = pickle.load(f)
VN, L, H = matrixs.shape
matrixs = np.transpose(matrixs, (1, 0, 2))
VN, L, H = matrixs.shape
with ProcessPoolExecutor(max_workers=config.num_workers) as executor:
futures = [
executor.submit(process_task, i, name, weights_file_name, H, L, config)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment