Skip to content

Commit

Permalink
try to fix multi-gpu
Browse files Browse the repository at this point in the history
Signed-off-by: YunLiu <[email protected]>
  • Loading branch information
KumoLiu committed Sep 30, 2024
1 parent 06251ea commit 007fcdf
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion models/vista3d/configs/multi_gpu_train.json
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
"shuffle": false
},
"validate#dataloader#sampler": "@validate#sampler",
"validate#evaluator#val_handlers": "$@validate#handlers[: -2 if dist.get_rank() > 0 else None]",
"validate#evaluator#val_handlers": "$@validate#handlers[: -3 if dist.get_rank() > 0 else None]",
"initialize": [
"$import torch.distributed as dist",
"$dist.is_initialized() or dist.init_process_group(backend='nccl')",
Expand Down

0 comments on commit 007fcdf

Please sign in to comment.