loadWeights

open override fun loadWeights(modelDirectory: File, loadOptimizerState: Boolean)

Loads variable data from .txt files.

Parameters

modelDirectory

Path to directory with TensorFlow graph and variable data.

loadOptimizerState

Loads optimizer internal variables data, if true.

Throws

FileNotFoundException

If file with weights is not found.