Fix optimizer step counter bug

5 years ago · 4419881f2f
parent 19334b4f05
commit 4419881f2f
1 changed files with 2 additions and 1 deletions
--- a/mindspore/ccsrc/optimizer/optimizer.h
+++ b/mindspore/ccsrc/optimizer/optimizer.h
@ -184,7 +184,8 @@ class Optimizer : public std::enable_shared_from_this<Optimizer> {
          }
        }
      };
-      use_profile ? (WITH(MsProfile::GetProfile()->Lap(counter++)) run_runc) : run_runc();
+      use_profile ? (WITH(MsProfile::GetProfile()->Lap(counter)) run_runc) : run_runc();
+      counter++;

      if (run_only_once_) {
        break;