!2841 Fix optimizer step counter bug

Merge pull request !2841 from Kang/master
5 years ago · d0d877a4ea
parent a342a615a8 4419881f2f
commit d0d877a4ea
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;