fixex imports

2026-09-12 12:40:20 +08:00 · 2019-04-21 13:12:42 -04:00
parent e9bcbc2318
commit 398b709b76
19 changed files with 1 additions and 1 deletions
@@ -0,0 +1,23 @@
+import torch
+
+class ModelHooks(torch.nn.Module):
+    def on_batch_start(self):
+        pass
+
+    def on_batch_end(self):
+        pass
+
+    def on_epoch_start(self):
+        pass
+
+    def on_epoch_end(self):
+        pass
+
+    def on_pre_performance_check(self):
+        pass
+
+    def on_post_performance_check(self):
+        pass
+
+    def should_stop_epoch(self, data_batch):
+        return False