Skip to content

Commit

Permalink
revert vista3d change
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 e6f21ba commit 8544181
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 4 deletions.
3 changes: 1 addition & 2 deletions models/vista3d/configs/metadata.json
Original file line number Diff line number Diff line change
@@ -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",
Expand Down
4 changes: 2 additions & 2 deletions models/vista3d/configs/multi_gpu_train.json
Original file line number Diff line number Diff line change
Expand Up @@ -17,15 +17,15 @@
},
"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",
"even_divisible": false,
"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')",
Expand Down

0 comments on commit 8544181

Please sign in to comment.