Browse Source

!2841 Fix optimizer step counter bug

Merge pull request !2841 from Kang/master
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
d0d877a4ea
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/ccsrc/optimizer/optimizer.h

+ 2
- 1
mindspore/ccsrc/optimizer/optimizer.h View File

@@ -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;


Loading…
Cancel
Save