Richie_C:

I had a problem with the models with neurons lower than 16 in a layer.


The more the hidden layer the more the number of parameters to learn (it depends on what kind of layers you have used). Higher the number of parameters, higher the memory required.
Please share a standalone code for further support. Thanks!