In that case either self.testing is still set to False (it seems to be set during initialization and I don’t know, why the internal self.training argument isn’t used) or self.final_activation is None.
Could you check both of these attributes and make sure they are set appropriately?