HyperGAN
Search…
Train Hook
Train hooks provide training events and loss modification to trainers.

Access

1
gan.trainer.train_hooks # => [...]
Copied!
Train hooks are setup and invoked by the trainer.

Events

Override these methods to change the train loop
1
before_step(step, feed_dict)
2
after_step(step, feed_dict)
3
after_create()
4
gradients(d_grads, g_grads)
Copied!

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 modified 1yr ago