Skip to content
This repository has been archived by the owner on Dec 21, 2023. It is now read-only.

Need the ability to set a smaller batch size for image classifier & image similarity #26

Closed
srikris opened this issue Dec 9, 2017 · 2 comments
Assignees

Comments

@srikris
Copy link
Contributor

srikris commented Dec 9, 2017

The feature extractor used image classifier and image similarity sets the batch size to 512 and we should expose that parameter all the way up to the image classifier and image similarity.

@gustavla
Copy link
Collaborator

gustavla commented Dec 9, 2017

If we do this, it would be good to expose this for object detection too to allow low-memory GPUs. However, it is harder since it doesn't use a stopping criterion based on a validation set and instead a heuristic to set the number of iterations. This heuristic would get more complicated.

@TobyRoseman
Copy link
Collaborator

This is fixed in beta 1 of version 5.0

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

No branches or pull requests

4 participants