DistributedDataParallel (DDP) RuntimeError: The size of tensor a (3) must match the size of tensor b (0) at non-singleton dimension 1 #21
Description
ns-train set n_gpus = 4
--machine.num-gpus $n_gpus
DistributedDataParallel (DDP) RuntimeError: The size of tensor a (3) must match the size of tensor b (0) at non-singleton dimension 1