Project-MONAI / Project-MONAI/model-zoo
Can't increase bs for Vista3d bundle finetuning
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 338
- Forks
- 106
- PR merge metrics
- No merged PRs in 30d
Description
Describe the bug
I am using vista3d fine tunning for a new task where I have 3 labels. It runs fine with bs=1 and it uses 8GB. I have a 48 GB gpu so I tried increasing the BS but I get an error. I also tried increasing the number of patched per image and got the same error
To Reproduce
download vista3d bundle
- change number of classes to 3
- change dataset json to your data as liver tumor dataset
- run fine tuning, it would run fine
- now change the bs to 2, rerun you will get an error
Expected behavior
I should be able to increase the BS >1 and make use of my GPU memory
Additional context
the GPU utilization is great it is always 90%+
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Download the Vista3D bundle and reproduce the fine-tuning flow with the reported three-label dataset setup. Compare the run with bs=1 against bs=2 and increased patches per image, recording the missing error details. Done means identifying why values above one fail and confirming fine-tuning runs with a larger batch size.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100