Open artempimushkin opened 5 years ago
Same issue here with V100 16GB GPU
It's just Out Of Memory of your GPU. You should modified the model to reduce memory (GPU) costs.
The code read all of the dataset once, so the memory costs very high. You can use light version of UGATIT as the README file said or use another GPU with high memory.
16GB GPU is dead ? kill me please.
Hey! I'm trying to use 512px images instead of 256px for better quality. I changed
default
flag from 256 to 512 in main.py -parser.add_argument('--img_size', type=int, default=512, help='The size of image')
and changedself.load_size == 256
toself.load_size == 512
in functionimage_processing()
in utils.py. But the script crashes (I'm using RTX 2080Ti 11GB with --light) with the following output: