How much memory does （train）test need（for CUFED5）？

[ximu1211]

How much memory does （train）test need（for CUFED5）？
Looking forward to your reply！

[Tengfei Wang]

Hi,
It does not consume much memory when inference. It takes ~ 3000M when testing on CUFED5.
For training, we set the patch size to 160, and adjust the batch size to utilize as much GPU memory as possible.