updated docs

This commit is contained in:
William Falcon
2019-07-25 11:52:54 -04:00
parent 600c755460
commit d272f29c88
3 changed files with 59 additions and 14 deletions
@@ -64,14 +64,6 @@ class LightningModule(GradInformation, ModelIO, ModelHooks):
"""
raise NotImplementedError
def loss(self, *args, **kwargs):
"""
Expand model_out into your components
:param model_out:
:return:
"""
raise NotImplementedError
def summarize(self):
model_summary = ModelSummary(self)
print(model_summary)