def METHOD_NAME(self):
    super().METHOD_NAME()
    for mod in self.project_features.children():
        if isinstance(mod, nn.Linear):
            mod.METHOD_NAME()
    if self.decoder is not None:
        self.decoder.METHOD_NAME()