cuda out of memory

[mouawiya]

i've tried the code on a computer with graphic card GTX 1050 Ti and i still got the Cuda out of memory error, when i tried to lowed the batch size, it worked for a moment then stopped as it is showed in the picture below

[tanay-o]

Hey! got the same error, try reducing the batch size when you train your dataset with the --batch_size parameter (default 4), that worked for me.