Thanks, I was hoping there would be a simple way. Thanks for letting me know
Probably the best way would be the training step as I am not sure how many layers require regularization. I want the solution to be generic and not specific