Could you provide a re-implement with one GPU?
Beyondzjl opened this issue · comments
You can do that by change the --nproc_per_node=8 to --nproc_per_node=1
how to choose a specific gpu
A PyTorch implementation of MAGE: MAsked Generative Encoder to Unify Representation Learning and Image Synthesis
Beyondzjl opened this issue · comments
You can do that by change the --nproc_per_node=8 to --nproc_per_node=1
how to choose a specific gpu