-
Notifications
You must be signed in to change notification settings - Fork 7
Open
Description
Hi! I am having this issue when training with batch size > 1:
tensorflow.python.framework.errors_impl.InvalidArgumentError: 2 root error(s) found.
(0) Invalid argument: Cannot add tensor to the batch: number of elements does not match. Shapes are: [tensor]: [119130,1], [batch]: [80000,1]
[[node IteratorGetNext (defined at train.py:117) ]]
(1) Invalid argument: Cannot add tensor to the batch: number of elements does not match. Shapes are: [tensor]: [119130,1], [batch]: [80000,1]
[[node IteratorGetNext (defined at train.py:117) ]]
[[Shape/_4]]
0 successful operations.
0 derived errors ignored. [Op:__inference_train_function_14011]
Function call stack:
train_function -> train_function
Metadata
Metadata
Assignees
Labels
No labels
