diff --git a/models/vista3d/configs/mgpu_evaluate.json b/models/vista3d/configs/mgpu_evaluate.json index 178b881e..f220c6f2 100644 --- a/models/vista3d/configs/mgpu_evaluate.json +++ b/models/vista3d/configs/mgpu_evaluate.json @@ -14,7 +14,7 @@ "shuffle": false }, "validate#dataloader#sampler": "@validate#sampler", - "validate#evaluator#val_handlers": "$@validate#handlers[: -2 if dist.get_rank() > 0 else None]", + "validate#handlers#1#_disabled_": "$dist.get_rank() > 0", "initialize": [ "$import torch.distributed as dist", "$dist.is_initialized() or dist.init_process_group(backend='nccl')",