diff --git a/models/vista3d/configs/metadata.json b/models/vista3d/configs/metadata.json index 00cb6e09..30904a97 100644 --- a/models/vista3d/configs/metadata.json +++ b/models/vista3d/configs/metadata.json @@ -1,8 +1,7 @@ { "schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20240725.json", - "version": "0.5.3", + "version": "0.5.2", "changelog": { - "0.5.3": "fix multi-gpu issue", "0.5.2": "fix eval issue", "0.5.1": "add description for zero-shot and upate eval", "0.5.0": "update json file link and add test", diff --git a/models/vista3d/configs/multi_gpu_train.json b/models/vista3d/configs/multi_gpu_train.json index f323b026..9fcc961e 100644 --- a/models/vista3d/configs/multi_gpu_train.json +++ b/models/vista3d/configs/multi_gpu_train.json @@ -17,7 +17,7 @@ }, "train#dataloader#sampler": "@train#sampler", "train#dataloader#shuffle": false, - "train#trainer#train_handlers": "$@train#handlers[: -2 if dist.get_rank() > 0 else None]", + "train#trainer#train_handlers": "$@train#handlers[: -1 if dist.get_rank() > 0 else None]", "validate#sampler": { "_target_": "DistributedSampler", "dataset": "@validate#dataset", @@ -25,7 +25,7 @@ "shuffle": false }, "validate#dataloader#sampler": "@validate#sampler", - "validate#evaluator#val_handlers": "$@validate#handlers[: -3 if dist.get_rank() > 0 else None]", + "validate#evaluator#val_handlers": "$@validate#handlers[: -2 if dist.get_rank() > 0 else None]", "initialize": [ "$import torch.distributed as dist", "$dist.is_initialized() or dist.init_process_group(backend='nccl')",