remove shortform (#1509)

rwth-i6 · Mar 22, 2024 · 85ce153 · 85ce153
1 parent 870826b
commit 85ce153
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/torch_avg_checkpoints.py b/tools/torch_avg_checkpoints.py
@@ -111,7 +111,7 @@ def merge_checkpoints(in_ckpts: Sequence[str], out_ckpt: str, extra_state: Optio
 
     # Average
     for k in out_model_state:
-        out_model_state[k] /= out_model_state_num[k]
+        out_model_state[k] = out_model_state[k] / out_model_state_num[k]
 
     if extra_state:
         out_state.update(extra_state)