Merge pull request #55 from williamFalcon/continue

add training restore
This commit is contained in:
William Falcon
2019-08-07 09:02:16 -04:00
committed by GitHub
8 changed files with 154 additions and 9 deletions
+8
View File
@@ -2,6 +2,14 @@ import torch
class ModelHooks(torch.nn.Module):
def on_sanity_check_start(self):
"""
Called before starting validate
:return:
"""
pass
def on_batch_start(self, data_batch):
pass