Skip to content

Instantly share code, notes, and snippets.

What would you like to do?
# Modified from:
class StoreHook(HookCallback):
def on_train_begin(self, **kwargs):
self.hists = []
def hook(self, m, i, o):
return o
def on_batch_end(self, train, **kwargs):
if (train):
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.