You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

loss_hooks.py 110B

3 months ago
12345678
  1. loss_hooks = []
  2. def add_to_loss_hooks(fn):
  3. loss_hooks.append(fn)
  4. def get_hooks():
  5. return loss_hooks