Train Hook
Train hooks provide training events and loss modification to trainers.
https://github.com/HyperGAN/HyperGAN/tree/master/hypergan/train_hooks
Access
Train hooks are setup and invoked by the trainer.
Events
Override these methods to change the train loop
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