Tried to train an image classification model using the Image Classification API. The value set for batchSize parameter is 300. Meanwhile then number of data instances in the test set is 182.
What happened?
No evaluation takes place. 0 batches are processed.
What did you expect?
The model to train and for it to evaluate the number of instances provided. In this case since the number of data instances is less than the amount set for the batchSize parameter, it would process 1 batch instead of 0.
This happens because currently, a batch is only processed if it has number of samples, and any incomplete batch is skipped. @codemzs is working on changing this.
System information
Issue
Tried to train an image classification model using the Image Classification API. The value set for
batchSize
parameter is 300. Meanwhile then number of data instances in the test set is 182.No evaluation takes place. 0 batches are processed.
The model to train and for it to evaluate the number of instances provided. In this case since the number of data instances is less than the amount set for the
batchSize
parameter, it would process 1 batch instead of 0.The model to evaluate
Source code / logs
Pipeline:
Output:
When the
batchSize
is set equal to the number of rows (in this case 182), this is the output: