Pretrained model for reconstruction network?

[nirajpandkar]

Hello Shunwang Gong,

I am trying to train the reconstruction model on a Tesla K40 (12GB memory) but unfortunately running out of GPU memory mid-way training. I'm training the network with the CoMA dataset as given in the code (more than 20k faces).

Could you help me with some strategies to combat this issue? I do have a cluster of GPUs available at my disposal but since the paper doesn't talk about any resources required to train the networks, I thought I would ask you first.

Or perhaps if it's possible, could you please share the pretrained model for reconstruction? I would like to use the encoder network on some of my in-house dataset.

Thank you,
Niraj Pandkar