Skip to content

Commit 95187ba

Browse files
committed
Fix finetuned checkpoint path
Signed-off-by: heyufan1995 <heyufan1995@gmail.com>
1 parent 588b299 commit 95187ba

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

vista3d/cvpr_workshop/README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ pip install -r requirements.txt
2323
```
2424

2525
# Training
26-
Download the challenge subsets finetuned [checkpoint](https://drive.google.com/file/d/1r2KvHP_30nHR3LU7NJEdscVnlZ2hTtcd/view?usp=sharing) or VISTA3D original [checkpoint]((https://drive.google.com/file/d/1DRYA2-AI-UJ23W1VbjqHsnHENGi0ShUl/view?usp=sharing)). Generate a json list that contains your traning data and update the json file path in the script.
26+
Download the challenge subsets finetuned [checkpoint](https://drive.google.com/file/d/1hQ8imaf4nNSg_43dYbPSJT0dr7JgAKWX/view?usp=sharing) or VISTA3D original [checkpoint]((https://drive.google.com/file/d/1DRYA2-AI-UJ23W1VbjqHsnHENGi0ShUl/view?usp=sharing)). Generate a json list that contains your traning data and update the json file path in the script.
2727
```
2828
torchrun --nnodes=1 --nproc_per_node=8 train_cvpr.py
2929
```
@@ -35,5 +35,6 @@ We provide a Dockerfile to satisfy the challenge format. For more details, refer
3535
docker build -t vista3d:latest .
3636
docker save -o vista3d.tar.gz vista3d:latest
3737
```
38+
You can also directly run `predict.sh`. Download the finetuned checkpoint and modify the `--model=/your_downloaded_checkpoint'.
3839

3940

0 commit comments

Comments
 (0)