Skip to content

S
seminar-breakout
Commit 56dd2390 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix #1057

parent 3ad0ed00
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tensorpack/dataflow/common.py
View file @ 56dd2390
...@@ -161,9 +161,9 @@ class BatchData(ProxyDataFlow): ...@@ -161,9 +161,9 @@ class BatchData(ProxyDataFlow):
else: else:
result.append(BatchData._batch_numpy(data_list)) result.append(BatchData._batch_numpy(data_list))
elif isinstance(first_dp, dict): elif isinstance(first_dp, dict):
result = [] result = {}
for key in first_dp.keys(): for key in first_dp.keys():
data_list = [x[k] for x in data_holder] data_list = [x[key] for x in data_holder]
if use_list: if use_list:
result[key] = data_list result[key] = data_list
else: else:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment