timekiller
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix albert loss not found issue

This commit is contained in:
zhengjun10 2021-12-15 16:00:09 +08:00
parent daae93cff3
commit 8394ee9878
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/lite/include/train/train_cfg.h
View File

@ -64,9 +64,9 @@ class TrainCfg {
this->accumulate_gradients_ = rhs.accumulate_gradients_;
return *this;
}
std::vector<std::string> loss_name_; /**< Set part of the name that identify a loss kernel */
MixPrecisionCfg mix_precision_cfg_; /**< Mix precision configuration */
bool accumulate_gradients_ = false; /**< If true gardents are accmulated and can be read by GetGradients */
std::vector<std::string> loss_name_ = {"loss_fct"}; /**< Set part of the name that identify a loss kernel */
MixPrecisionCfg mix_precision_cfg_; /**< Mix precision configuration */
bool accumulate_gradients_ = false; /**< If true gardents are accmulated and can be read by GetGradients */
};
} // namespace lite