vess2ret
vess2ret copied to clipboard
Error when next batch is to be loaded
index_array, _, current_batch_size = next(self.index_generator) ValueError: need more than 1 value to unpack
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.