deepxde.optimizers.tensorflow package

Submodules

deepxde.optimizers.tensorflow.optimizers module

deepxde.optimizers.tensorflow.optimizers.get(optimizer, learning_rate=None, decay=None)[source]

Retrieves a Keras Optimizer instance.

deepxde.optimizers.tensorflow.optimizers.is_external_optimizer(optimizer)[source]

deepxde.optimizers.tensorflow.tfp_optimizer module

TensorFlow interface for TensorFlow Probability optimizers.

deepxde.optimizers.tensorflow.tfp_optimizer.lbfgs_minimize(trainable_variables, build_loss, previous_optimizer_results=None)[source]

TensorFlow interface for tfp.optimizer.lbfgs_minimize.

Parameters:
  • trainable_variables – Trainable variables, also used as the initial position.

  • build_loss – A function to build the loss function expression.

  • previous_optimizer_results

Module contents

deepxde.optimizers.tensorflow.get(optimizer, learning_rate=None, decay=None)[source]

Retrieves a Keras Optimizer instance.

deepxde.optimizers.tensorflow.is_external_optimizer(optimizer)[source]