提交 b67d78e7 编写于 作者: W wangyanfei01 提交者: Yu Yang

fix bug: if test and sparse_remote_update can not co-exsit, crash trainer if necessary

上级 573cdb60
......@@ -222,6 +222,12 @@ void Trainer::init(const std::shared_ptr<TrainerConfigHelper>& config,
DataProvider::create(config_->getTestDataConfig(), *config_, gpuData));
}
if (testDataProvider_) {
if (config_->getOptConfig().use_sparse_remote_updater()) {
LOG(FATAL) << "It's prohibited to set sparse_remote_update "
<< "in some layers if testing will be under going "
<< "in the middle of training. You can do testing "
<< "within separate process.";
}
createTester();
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册