提交 80aef42e 编写于 作者: S sneaxiy

test commit for reverting

上级 a5a373f4
...@@ -74,7 +74,9 @@ class OrderedMultiDeviceLoDTensorBlockingQueue { ...@@ -74,7 +74,9 @@ class OrderedMultiDeviceLoDTensorBlockingQueue {
public: public:
OrderedMultiDeviceLoDTensorBlockingQueue(size_t capacity, OrderedMultiDeviceLoDTensorBlockingQueue(size_t capacity,
bool speed_test_mode = false) bool speed_test_mode = false)
: capacity_(capacity), speed_test_mode_(speed_test_mode) {} : capacity_(capacity), speed_test_mode_(speed_test_mode) {
LOG(INFO) << "Use keep_order=True DataLoader";
}
~OrderedMultiDeviceLoDTensorBlockingQueue() { ~OrderedMultiDeviceLoDTensorBlockingQueue() {
VLOG(10) << "Destruct OrderedMultiDeviceLoDTensorBlockingQueue"; VLOG(10) << "Destruct OrderedMultiDeviceLoDTensorBlockingQueue";
......
...@@ -89,7 +89,7 @@ class DataLoader(object): ...@@ -89,7 +89,7 @@ class DataLoader(object):
return_list=False, return_list=False,
use_multiprocess=False, use_multiprocess=False,
drop_last=True, drop_last=True,
keep_order=True): keep_order=False):
""" """
Create a DataLoader object for loading data from Python generator. Create a DataLoader object for loading data from Python generator.
Data would be prefetched using Python thread and be pushed Data would be prefetched using Python thread and be pushed
...@@ -633,7 +633,7 @@ class GeneratorLoader(DataLoaderBase): ...@@ -633,7 +633,7 @@ class GeneratorLoader(DataLoaderBase):
iterable=True, iterable=True,
return_list=False, return_list=False,
drop_last=True, drop_last=True,
keep_order=True): keep_order=False):
self._tensor_reader = None self._tensor_reader = None
self._places = None self._places = None
self._thread = None self._thread = None
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册