Fixed unbatching of dict

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens 2019-07-11 17:09:21 +02:00
parent 1474f30744
commit 8d1d78bdb6
1 changed files with 2 additions and 0 deletions

View File

@ -410,6 +410,8 @@ def _ssd_evaluate_unbatch_dict(glob_string: str) -> tuple:
nr_keys = len(batched.keys())
unbatched_dict = tuple([[] for _ in range(nr_keys)])
batched = batched.values()
for i in range(nr_keys):
value = batched[i]
unbatched_dict[i].extend(value)