Batch size and other hyperparameters

ManishSoni1908 / Mobilenet-ssd-keras

A Keras port of Single Shot MultiBox Detector as Mobilenet as a Backbone

62 stars 24 forks source link

Closed davideCremona closed 4 years ago

davideCremona commented 4 years ago

Hi, I'm experimenting with your repository but when I start training I get OOM on NVIDIA RTX 2080.

I think that maybe it's the batch size that is too much but I don't see any place where I can set this type of parameters.

Can you help me please?

davideCremona commented 4 years ago

It's a command line parameter, my fault.