提交 1cbe4bf2 编写于 作者: 文幕地方's avatar 文幕地方

add todo

上级 23eb335d
...@@ -24,6 +24,7 @@ class DictCollator(object): ...@@ -24,6 +24,7 @@ class DictCollator(object):
""" """
def __call__(self, batch): def __call__(self, batch):
# todo:support batch operators
data_dict = defaultdict(list) data_dict = defaultdict(list)
to_tensor_keys = [] to_tensor_keys = []
for sample in batch: for sample in batch:
...@@ -43,6 +44,7 @@ class ListCollator(object): ...@@ -43,6 +44,7 @@ class ListCollator(object):
""" """
def __call__(self, batch): def __call__(self, batch):
# todo:support batch operators
data_dict = defaultdict(list) data_dict = defaultdict(list)
to_tensor_idxs = [] to_tensor_idxs = []
for sample in batch: for sample in batch:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册