# System Requirements to train a Image Recognition Neural Network

I have 6000 Images to be trained on a Neural Network.

My current PC Specs :-

32GB RAM, i5 2 core Processor, Standard GPU (No work going on GPU), 1TB Hard Disk

My Neural Network Specs :-

3000 Epoch Size, 6 Batch Size

Process is getting killed with these specifications.

How much Hardware is required to trained it well? What would be the ideal Batch Size and Epoch Size?

• possible flavors of resource constraints : CPU (or gpu but not in this case), memory bandwidth, total memory, disc IO access ... you need to learn how to monitor your system to see which is the weakest link ... lower number of images to 1000 and see what which of above is starting to sweat – Scott Stensland Aug 24 '18 at 21:19

Try creating your model (or a similar one) using Keras instead of pure Tensorflow, and then using model.summary on the compiled model. This will show you a nice overview of the number of weights in each layer.