We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 27fa422 commit 1911bc4Copy full SHA for 1911bc4
QEfficient/finetune/utils/train_utils.py
@@ -308,6 +308,7 @@ def train(
308
309
if train_config.task_type == "seq_classification":
310
accuracy = acc_helper.compute()
311
+ acc_helper.reset()
312
if train_config.enable_ddp:
313
dist.all_reduce(accuracy, op=dist.ReduceOp.SUM)
314
accuracy /= dist.get_world_size()
0 commit comments