Skip to content

Commit 1911bc4

Browse files
meetkumaquic-meetkuma
meetkuma
authored andcommitted
Fixed train accuracy computation
Signed-off-by: Meet Patel <[email protected]>
1 parent 27fa422 commit 1911bc4

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

QEfficient/finetune/utils/train_utils.py

+1
Original file line numberDiff line numberDiff line change
@@ -308,6 +308,7 @@ def train(
308308

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

0 commit comments

Comments
 (0)