Train Hook

Train hooks provide training events and loss modification to trainers.

https://github.com/HyperGAN/HyperGAN/tree/master/hypergan/train_hooks

Access

gan.trainer.train_hooks # => [...]

Train hooks are setup and invoked by the trainer.

Events

Override these methods to change the train loop

before_step(step, feed_dict)
after_step(step, feed_dict)
after_create()
gradients(d_grads, g_grads)

before_step(feed_dict)

after_step(feed_dict)

Executed before/after the step takes place. feed_dict is what is being sent to the graph during the training step.

after_create()

Ran after the trainer is created.

gradients(d_grads, g_grads)

Refines the gradients before they are applied to the optimizer.

Last updated