From 4798cc665906d7aecfcdc3361d2e07656adca99c Mon Sep 17 00:00:00 2001 From: ldwang Date: Sun, 28 Apr 2024 13:53:53 +0800 Subject: [PATCH] Update trainer_v1.py --- flagai/trainer_v1.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flagai/trainer_v1.py b/flagai/trainer_v1.py index adbf5618..94ab0fc2 100755 --- a/flagai/trainer_v1.py +++ b/flagai/trainer_v1.py @@ -876,7 +876,7 @@ def train_step_pytorch(self, optimizer.zero_grad() else: optimizer.step() - # optimizer.zero_grad() + optimizer.zero_grad() self.accumulate_count = 0 else: self.accumulate_count += 1