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.

Components - Previous

Optimizer

Adversarial Norm

Last modified 2yr ago