Skip to content

Commit 977f881

Browse files
committed
Fixed train accuracy computation
Signed-off-by: Meet Patel <[email protected]>
1 parent 621398f commit 977f881

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

QEfficient/finetune/utils/train_utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -309,6 +309,7 @@ def train(
309309

310310
if train_config.task_type == "seq_classification":
311311
accuracy = acc_helper.compute()
312+
acc_helper.reset()
312313
if train_config.enable_ddp:
313314
dist.all_reduce(accuracy, op=dist.ReduceOp.SUM)
314315
accuracy /= dist.get_world_size()

0 commit comments

Comments
 (0)