Newbeeer / L_DMI

Code for NeurIPS 2019 Paper, "L_DMI: An Information-theoretic Noise-robust Loss Function"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

For Clothing1M dataset, how many GPU memory are needed?

guixianjin opened this issue · comments

Hi! I try to run the code for Clothing1M. But error report: GPU memory is not enough. I use RTX 2080Ti, about 10G GPU memory. So, I want to know if it's the GPU‘ memory too small.

Hi! We use 6 GPU during training (batch size=256), with 12G memory on each GPU.

OKay. Very thanks!