Closed Dawars closed 2 years ago
Made some fixes to be able to run on a single machine with a single GPU.
Use with --num_gpus 1 --num_nodes 1 for single GPU
--num_gpus 1 --num_nodes 1
Made some fixes to be able to run on a single machine with a single GPU.
Use with
--num_gpus 1 --num_nodes 1
for single GPU