Isn’t there any attribute of the network itself that can be accessed to identify it?
When we do model.train() or model.eval() I assume something in the model must change so batchnorm and dropout layers knows how to behave in the inference pass.
Is this correct?