Add callbacks (#7315)

* Add `on_train_start()` callback

* Update

* Update
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-04-06 17:23:34 +02:00
1 parent 32661f75ac
commit 245d6459a9
3 files changed
+12

No files matched your search

+4
View File
@@ -84,6 +84,10 @@ class Loggers():
else:
self.wandb = None
def on_train_start(self):
# Callback runs on train start
pass
def on_pretrain_routine_end(self):
# Callback runs on pre-train routine end
paths = self.save_dir.glob('*labels*.jpg') # training labels