remove shortform (rwth-i6#1509)

JackTemaki · Mar 22, 2024 · 0bec0cc · 0bec0cc
1 parent 8f5d2a5
commit 0bec0cc
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)