Thanks a lot 
I had another doubt. Would really appreciate the help
If my last layer has one output, i.e., my target is to train a model to get only one class(example: cat)…which loss function would be adequate for it? I was using MSE but it doesn’t converge. It goes past the target value if more data is used.