CUDA out of memory during training

[weijiawang96]

Dear Authors,

I tried to train using your given dataset 0 (command: python Train.py --dataset 0). The available memory is around 10GB. However, the training cannot start as it keeps saying CUDA out of memory.

I wonder if 10GB is enough? I am curious how much memory you used on your GeForce RTX 2080 when you trained your model. Or how can we use multiple GPUs to train? Alternatively, is there a way to change the batch size?

Thank you for your help! I appreciate it.