Open divyagaddipati opened 6 years ago
Hi!
What is the keras version you are using?
2.1.5
I am sorry, this code was developed with keras 1. If you wish to use keras 2 you will probably need to make some changes.
In keras 2, the index_generator only returns the index_array. I guess that you can easily get the current_batch_size by getting the lenght of the index_array!
Okay, changed it accordingly. Thanks.
indexarray, , current_batch_size = next(self.index_generator) ValueError: need more than 1 value to unpack