Got an AssertionError when run the train_SAM_box.py

wangsssky / SonarSAM

Segment Anything Model, SAM, Sonar images

Apache License 2.0

56 stars 5 forks source link

Got an AssertionError when run the train_SAM_box.py #7

Open monxxcn opened 8 months ago

monxxcn commented 8 months ago

Hello, I encountered an error when attempting to fine-tune SonarSAM using box prompts. I followed your guidelines precisely while running the program. What could be the potential issues？截屏2024-03-06 19 01 49 截屏2024-03-06 19 02 24

monxxcn commented 8 months ago

I ran the code with cuda 11.8, pytorch 2.0.1 and torchvision 0.15.2

monxxcn commented 8 months ago

It seems that the inconsistent data distribution which is a result of distributed training. Is there a recommended solution to address this issue?