I have built a neural net with a custom architecture, as designed below. No I wanted to perform a hyperparameter optimization in order to find the optimal learning rate, batch s