search

amazon-science / gan-control

This package provides a pythorch implementation of "GAN-Control: Explicitly Controllable GANs", ICCV 2021.
Apache License 2.0
137 stars 23 forks source link

cuda out of memory,how can i reduce batch size #9

Closed pfeducode closed 2 years ago

pfeducode commented 2 years ago

When I started training, I threw an error. cuda out of memory. I try to reduce the batchsize from 16 to 4, but another error will be thrown, mini_ Batch cannot be divided. How can I start training on two or oneRTX3090. 1.batch size=16, cuda out of memory 1

  1. i reduce batch size from 16 to 4 2 3