Fixed unbatching of dict
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
@ -410,6 +410,8 @@ def _ssd_evaluate_unbatch_dict(glob_string: str) -> tuple:
|
|||||||
nr_keys = len(batched.keys())
|
nr_keys = len(batched.keys())
|
||||||
unbatched_dict = tuple([[] for _ in range(nr_keys)])
|
unbatched_dict = tuple([[] for _ in range(nr_keys)])
|
||||||
|
|
||||||
|
batched = batched.values()
|
||||||
|
|
||||||
for i in range(nr_keys):
|
for i in range(nr_keys):
|
||||||
value = batched[i]
|
value = batched[i]
|
||||||
unbatched_dict[i].extend(value)
|
unbatched_dict[i].extend(value)
|
||||||
|
|||||||
Reference in New Issue
Block a user