williamSYSU / TextGAN-PyTorch

TextGAN is a PyTorch framework for Generative Adversarial Networks (GANs) based text generation models.
MIT License
892 stars 205 forks source link

Can you tell me the GPU model and running time when running CatGaN? #50

Open xuefeng199507 opened 3 years ago

xuefeng199507 commented 3 years ago

My GPU model is 2080Ti . But it will raise an error:RuntimeError: CUDA out of memory. Tried to allocate 1.43 GiB (GPU 4; 10.76 GiB total capacity; 4.20 GiB already allocated; 147.69 MiB free; 4.28 GiB reserved in total by PyTorch) Can you tell me how to solve it? Can only reduce the batchsize?