Since Recorder tries to count the number of samples in a batch separately from those in a batch, it fails to get the batch size if the model input xs is not an array.
Possible solutions:
remove the sample-counting
add batchsize detection cases for other input types like Tuples and Dicts
I prefer 1. since the feature is not currently used, and it is not possible to exhaustively implement extra cases for batch size detection. Recorder should focus on counting the steps.
Since
Recorder
tries to count the number of samples in a batch separately from those in a batch, it fails to get the batch size if the model inputxs
is not an array.Possible solutions:
Tuple
s andDict
sI prefer 1. since the feature is not currently used, and it is not possible to exhaustively implement extra cases for batch size detection.
Recorder
should focus on counting the steps.