search

autodistill / autodistill-grounded-sam

GroundedSAM Base Model plugin for Autodistill
https://docs.autodistill.com
Apache License 2.0
44 stars 8 forks source link

batch size? #8

Open nihirv opened 9 months ago

nihirv commented 9 months ago

Hi, is there any way to pass a batch_size param to either the GroundedSAM constructor or model.label()? Training one-by-one is a bit slow and doesn't utilise my GPU memory is efficiently as it could

capjamesg commented 5 months ago

We do not support this at the moment. If you would like to submit a PR, we can review it!