the basic_generator has something wrong

[garyliu0816]

When you try to use basic_generator to generate data, the first time is right, the shape is

(32, 32, 32, 3) (32,)

But when you generate data again, the shape of data is

(32,32,3) (32,32,3)

It will cause the following function receive wrong inputs, such as

return seq.augment(images.transpose(0, 2, 3, 1)).transpose(0, 2, 3, 1)

The shape of "images" is determined by the basic_generator. Can you solve this problem?