search

tianrun-chen / SAM-Adapter-PyTorch

Adapting Meta AI's Segment Anything to Downstream Tasks with Adapters and Prompts
MIT License
850 stars 75 forks source link

about epoch number set and training time #76

Open Salaydin opened 2 months ago

Salaydin commented 2 months ago

I want to train vit-b segmentation based on this code. But It is too slow to train it. For example, the training time is about 21 hours when I set my epoches to 30. But my model cannot learn well when I set the eopches 30. How many epoches number did you set ? and how many hours did it take in such setting?

tianrun-chen commented 1 month ago

It depends on your dataset and task. The training time has been reported in the manuscript.