How any GPU's required and what shall be the GPU memory?

fyviezhao / M3D-VTON

Official code for ICCV2021 paper "M3D-VTON: A Monocular-to-3D Virtual Try-on Network"

166 stars 38 forks source link

How any GPU's required and what shall be the GPU memory? #2

Closed dev20111 closed 2 years ago

fyviezhao commented 2 years ago

Hi, currently the M3D-VTON supports single GPU / CPU training and testing. It would occupy ~1.5G of GPU memory for MTM and TFM, while for DRM the memory usage would increase to ~2G during inference.