Closed purijs closed 3 years ago
Try to use batch size = 3 or lower.
Hello, I'm having the same issue but using the COWC dataset that is used as reference on Github. Do you refer to the batch size in config_GAN.json? Thanks!
Yes, change it in the config_GAN.json file. Thanks.
I'm trying to train on 256x256 tiles, both train and test are 256x256. Also, I have a 16GB GPU still it runs out of memory