Problem "Resource exhausted: OOM when allocating tensor" about faceswap HOT 6 CLOSED

Open model.py in lib, try reduce the filter number

def Encoder():
    input_ = Input(shape=IMAGE_SHAPE)
    x = input_
    x = conv(128)(x)
    x = conv(256)(x)
    x = conv(512)(x)
    x = conv(1024)(x)
    x = Dense(ENCODER_DIM)(Flatten()(x))
    x = Dense(4 * 4 * 1024)(x)
    x = Reshape((4, 4, 1024))(x)
    x = upscale(512)(x)
    return Model(input_, x)

It works.
This is the link. Thanks deepfakes.

from faceswap.

Resource exhausted: OOM when allocating tensor with shape[3, 3, 128, 256]

OOM = Out of Memory. This means there is not enough graphics memory available to load the training data. It is possible that this is related to this issue,, which I will most probably be fixing tomorrow or the day after. (or, if you are capable, you could attempt to fix it on your own and send a pull request for us to merge)

But while that may be the case here, remember that a GTX660 has only 2GB of usable graphics memory in its default configuration. Most people who have had this running on GPU were using at least 4GB of graphics memory. It could therefore be possible that it is not related to that bug and that you do lack the memory required.

from faceswap.

It's not necessarily about Tensorflow. It's not necessarily about the pictures you are using. It's about the training model as well. There are a lot of factors to consider. Essentially, for this specific processing task, you need around 3-4GB of graphics memory. I'm unsure about specifying the size of the loaded training data; I would have to dive into Tensorflow and Keras in more detail. Perhaps this is something that can be improved in the future, but it looks like you might be stuck with CPU training.

from faceswap.

Thanks for explaining it.
Is there a method to specify the size of the loaded train data?
Even I load one pictures for training, it show me the OOM.
The Tensorflow-gpu has a minimum requirement for memory of video card (at least 4GB )?
Thanks.

from faceswap.

Also this guy had the problem as his card is having only 2Gb: https://www.reddit.com/r/deepfakes/comments/7mqob8/first_try_with_smaller_network/

from faceswap.

This issue was moved to deepfakes/faceswap-playground#13

from faceswap.