gwxie / Dewarping-Document-Image-By-Displacement-Flow-Estimation

Dewarping Document Image By Displacement Flow Estimation with Fully Convolutional Network
MIT License
166 stars 37 forks source link

GPU memory for training #15

Open Aiden0609 opened 1 year ago

Aiden0609 commented 1 year ago

Hi, splendid work you've done! I've been considering reproducing it but I don't know what kind of hardware conditions are recommended to successfully reproduce the training of networks, so could you tell me the GPU memory consumed for your training process? Tks. I really need to know this!