提交 8d8388c5 编写于 作者: T Tao Luo 提交者: GitHub

Merge pull request #686 from emailweixu/fix_test

Fix --job=test
...@@ -271,7 +271,9 @@ public: ...@@ -271,7 +271,9 @@ public:
void finishAsyncLoad() { void finishAsyncLoad() {
stopping_ = true; stopping_ = true;
taskReadySem_.post(); taskReadySem_.post();
asyncLoader_->join(); if (asyncLoader_) {
asyncLoader_->join();
}
} }
void setPending(bool pending) { pending_ = pending; } void setPending(bool pending) { pending_ = pending; }
......
...@@ -205,7 +205,7 @@ void Trainer::init(const std::shared_ptr<TrainerConfigHelper>& config, ...@@ -205,7 +205,7 @@ void Trainer::init(const std::shared_ptr<TrainerConfigHelper>& config,
(!IGradientMachineMode::dataMustInCpu(mode_, FLAGS_trainer_count)); (!IGradientMachineMode::dataMustInCpu(mode_, FLAGS_trainer_count));
dataProvider_ = dataProvider; dataProvider_ = dataProvider;
if (!dataProvider_ && config_->hasDataConfig()) { if (!dataProvider_ && config_->hasDataConfig() && !testing_) {
dataProvider_.reset(DataProvider::create(*config_, *config_, gpuData)); dataProvider_.reset(DataProvider::create(*config_, *config_, gpuData));
} }
if (!testDataProvider_) { if (!testDataProvider_) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册