mage
mage copied to clipboard
Could you provide a re-implement with one GPU?
You can do that by change the --nproc_per_node=8 to --nproc_per_node=1
how to choose a specific gpu