loss_hooks = [] def add_to_loss_hooks(fn): loss_hooks.append(fn) def get_hooks(): return loss_hooks