search

tianrun-chen / SAM-Adapter-PyTorch

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

batch size？ #26

Closed kaigelee closed 1 year ago

kaigelee commented 1 year ago

Batch size 只能为1吗？

yang-Michael commented 1 year ago

只要显存够，多少都行

tianrun-chen commented 1 year ago

You can use any batch size if the memory of GPU is sufficient