Giters

maum-ai / faceshifter

Unofficial PyTorch Implementation for FaceShifter (https://arxiv.org/abs/1912.13457)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

GPU Memory

superxi opened this issue · comments

commented

Hi :
I‘m wondering how much GPU memory is needed for training with batch size 1. I'm training with 2080Ti 11G, It always CUDA out of memory even when batch size is 1

commented

try ReLU(inplace=True)

commented

I have a same problem.
Did you solved this problem?