Source: vis/callbacks.py#L0
OptimizerCallback
Optimizer.minimizeで使用するコールバックを定義するための抽象クラスです。
OptimizerCallback.callback
callback(self, i, named_losses, overall_loss, grads, wrt_value)
この関数はoptimizer.minimizeの中で呼び出されます。
引数:
- i: 最適化の反復数
- named_losses:
(損失名, 損失値)
のタプルのリスト - overall_loss: 全ての重み付き損失
- grads:
wrt_value
に関する入力画像の勾配 - wrt_value: 現時点の
wrt_value
OptimizerCallback.on_end
on_end(self)
最適化プロセスの最後に呼び出されます。 この関数は通常、最適化の最後に使用していたリソースのクリーンアップやクローズに使われます。
最適化中の値をプリントするためのコールバックです。
Print.callback
callback(self, i, named_losses, overall_loss, grads, wrt_value)
Print.on_end
on_end(self)
GifGenerator
最適化画像のGIFを構築するためのコールバックです。
GifGenerator.__init__
__init__(self, path)
引数:
- path: 保存するGIFファイルのパス。
GifGenerator.callback
callback(self, i, named_losses, overall_loss, grads, wrt_value)
GifGenerator.on_end
on_end(self)