If I use the functional, that functional layer is not included, when I print out my model with print(self.model).  Is there a way to circumvent that ?  I find it quite useful to print out a model as a final check…