CUDA error: out of memory

BIT-DA / I2V-GAN

ACMMM2021 paper "I2V-GAN: Unpaired Infrared-to-Visible Video Translation"

https://dl.acm.org/doi/10.1145/3474085.3475445

MIT License

109 stars 22 forks source link

CUDA error: out of memory #13

Open xwhkkk opened 1 year ago

xwhkkk commented 1 year ago

Hello! thanks for sharing your work. I followed the instructions for running the python train.py, but it has the following error. Could you give me some suggestions? The dataroot I used is IRVI/triplet/traffic, and my GPU device is RTX3090.

zza1617 commented 1 year ago

reduce batch_size