-
Notifications
You must be signed in to change notification settings - Fork 1.8k
Closed
Description
Problem should lies at the following codes.
Lines 35 to 38 in 5a99f58
| # use the first key as main key to calculate the batch size | |
| batch_size = len(next(iter(data.values()))) | |
| for _ in range(batch_size): | |
| prog_bar.update() |
When running with classification, the first values (image_meta) is a data container with nested list, like DataContainer([[{meta1}, {meta2}, ... ]]), thus batch_size = 1 and is not desired.
I suggest using the length of the result as batch size, would it be good to all packages?
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels