From 85441812742de8d18313fa513799f00c9eb15495 Mon Sep 17 00:00:00 2001 From: YunLiu <55491388+KumoLiu@users.noreply.github.com> Date: Mon, 30 Sep 2024 21:50:11 +0800 Subject: [PATCH] revert vista3d change Signed-off-by: YunLiu <55491388+KumoLiu@users.noreply.github.com> --- models/vista3d/configs/metadata.json | 3 +-- models/vista3d/configs/multi_gpu_train.json | 4 ++-- 2 files changed, 3 insertions(+), 4 deletions(-) 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')",