Ngoc Anh Huynhon Stack Overflow • Recommends
On “Neural Network flatlines after one epoch”
…early. Suppose that your problem is learnable at all, there are many reasons for the for this behavior. These are at the top of my head:
Improper preprocessing of input:
Neural network prefers input with zero mean. E.g., if the input is all positive, it will restrict the weights to be updated in the same direction, which may not be desirable (link to this video).
Therefore, you may want to subtract the mean from all the images (e.g., subtract 127.5 from each of the 3 channels). Scaling to make unit standard deviation in each channel may also be helpful.
Generalization ability of the network:
The network is not complicated or deep enough for the task.
This is very easy to check. You can tra…