提交 a30f6aa9 编写于 作者: Y Yu Yang 提交者: GitHub

Merge pull request #1307 from reyoung/feature/fix_bug_in_dp2

Temporary disable async load data in PyDP2.
...@@ -54,7 +54,9 @@ before_install: ...@@ -54,7 +54,9 @@ before_install:
fi fi
- if [[ "$TRAVIS_OS_NAME" == "osx" ]]; then paddle/scripts/travis/before_install.osx.sh; fi - if [[ "$TRAVIS_OS_NAME" == "osx" ]]; then paddle/scripts/travis/before_install.osx.sh; fi
- if [[ "$JOB" == "PRE_COMMIT" ]]; then sudo ln -s /usr/bin/clang-format-3.8 /usr/bin/clang-format; fi - if [[ "$JOB" == "PRE_COMMIT" ]]; then sudo ln -s /usr/bin/clang-format-3.8 /usr/bin/clang-format; fi
- pip install numpy wheel protobuf sphinx recommonmark sphinx_rtd_theme virtualenv pre-commit requests==2.9.2 LinkChecker # Paddle is using protobuf 3.1 currently. Protobuf 3.2 breaks the compatibility. So we specify the python
# protobuf version.
- pip install numpy wheel 'protobuf==3.1' sphinx recommonmark sphinx_rtd_theme virtualenv pre-commit requests==2.9.2 LinkChecker
script: script:
- paddle/scripts/travis/main.sh - paddle/scripts/travis/main.sh
notifications: notifications:
......
...@@ -647,7 +647,7 @@ public: ...@@ -647,7 +647,7 @@ public:
DataBatch& gpuBatch = *batch; DataBatch& gpuBatch = *batch;
std::vector<Argument>& gpuArguments = gpuBatch.getStreams(); std::vector<Argument>& gpuArguments = gpuBatch.getStreams();
gpuArguments.resize(cpuArguments.size()); gpuArguments.resize(cpuArguments.size());
gpuBatch.setSize(size); gpuBatch.setSize(bsize);
for (size_t i = 0; i < headers_.size(); ++i) { for (size_t i = 0; i < headers_.size(); ++i) {
gpuArguments[i].resizeAndCopyFrom( gpuArguments[i].resizeAndCopyFrom(
cpuArguments[i], useGpu_, HPPL_STREAM_1); cpuArguments[i], useGpu_, HPPL_STREAM_1);
......
...@@ -201,7 +201,7 @@ def define_py_data_sources2(train_list, test_list, module, obj, args=None): ...@@ -201,7 +201,7 @@ def define_py_data_sources2(train_list, test_list, module, obj, args=None):
data.load_data_module = load_data_module data.load_data_module = load_data_module
data.load_data_object = load_data_object data.load_data_object = load_data_object
data.load_data_args = load_data_args data.load_data_args = load_data_args
data.async_load_data = True data.async_load_data = False
return data return data
define_py_data_sources( define_py_data_sources(
......
...@@ -19,7 +19,7 @@ model_config { ...@@ -19,7 +19,7 @@ model_config {
data_config { data_config {
type: "py2" type: "py2"
files: "train.list" files: "train.list"
async_load_data: true async_load_data: false
for_test: false for_test: false
load_data_module: "a" load_data_module: "a"
load_data_object: "c" load_data_object: "c"
...@@ -58,7 +58,7 @@ opt_config { ...@@ -58,7 +58,7 @@ opt_config {
test_data_config { test_data_config {
type: "py2" type: "py2"
files: "test.list" files: "test.list"
async_load_data: true async_load_data: false
for_test: true for_test: true
load_data_module: "b" load_data_module: "b"
load_data_object: "d" load_data_object: "d"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册