HyperGAN
Search
⌃K

Train Hook

Train hooks provide training events and loss modification to trainers.

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.