Hi everyone! Could you help me with this error?
ValueError: the input array must be have a shape == (.., ..,[ ..,] 3)), got (500, 333)

There are some values of batch size and number of images in the dataset that the training step works, but for other values I get this error…I don’t understand why this is happening…
Another problem that I remembered now is that when I increase the batch size, this error occurs:
RuntimeError: CUDA out of memory. Tried to allocate 148.00 MiB (GPU 0; 8.00 GiB total capacity; 5.60 GiB already allocated; 140.97 MiB free; 280.24 MiB cached)

In my understanding, there is memory on my video card to use a batch of this size … I don’t understand why the error occurs …
Best regards,
Matheus Santos