vess2ret icon indicating copy to clipboard operation
vess2ret copied to clipboard

Error when next batch is to be loaded

Open divyagaddipati opened this issue 6 years ago • 4 comments

index_array, _, current_batch_size = next(self.index_generator) ValueError: need more than 1 value to unpack

divyagaddipati avatar Mar 27 '18 11:03 divyagaddipati

Hi!

What is the keras version you are using?

costapt avatar Mar 27 '18 13:03 costapt

2.1.5

divyagaddipati avatar Mar 27 '18 13:03 divyagaddipati

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!

costapt avatar Mar 27 '18 14:03 costapt

Okay, changed it accordingly. Thanks.

divyagaddipati avatar Mar 28 '18 08:03 divyagaddipati