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

最適化中の値をプリントするためのコールバックです。


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)