gradgraph.optimization.tf package

Submodules

• gradgraph.optimization.tf.callbacks module
  • EarlyStoppingByThreshold
    • EarlyStoppingByThreshold.get_monitor_value()
    • EarlyStoppingByThreshold.model
    • EarlyStoppingByThreshold.on_batch_begin()
    • EarlyStoppingByThreshold.on_batch_end()
    • EarlyStoppingByThreshold.on_epoch_begin()
    • EarlyStoppingByThreshold.on_epoch_end()
    • EarlyStoppingByThreshold.on_predict_batch_begin()
    • EarlyStoppingByThreshold.on_predict_batch_end()
    • EarlyStoppingByThreshold.on_predict_begin()
    • EarlyStoppingByThreshold.on_predict_end()
    • EarlyStoppingByThreshold.on_test_batch_begin()
    • EarlyStoppingByThreshold.on_test_batch_end()
    • EarlyStoppingByThreshold.on_test_begin()
    • EarlyStoppingByThreshold.on_test_end()
    • EarlyStoppingByThreshold.on_train_batch_begin()
    • EarlyStoppingByThreshold.on_train_batch_end()
    • EarlyStoppingByThreshold.on_train_begin()
    • EarlyStoppingByThreshold.on_train_end()
    • EarlyStoppingByThreshold.set_model()
    • EarlyStoppingByThreshold.set_params()
  • ReduceLROnPlateau
    • ReduceLROnPlateau.in_cooldown()
    • ReduceLROnPlateau.model
    • ReduceLROnPlateau.on_batch_begin()
    • ReduceLROnPlateau.on_batch_end()
    • ReduceLROnPlateau.on_epoch_begin()
    • ReduceLROnPlateau.on_epoch_end()
    • ReduceLROnPlateau.on_predict_batch_begin()
    • ReduceLROnPlateau.on_predict_batch_end()
    • ReduceLROnPlateau.on_predict_begin()
    • ReduceLROnPlateau.on_predict_end()
    • ReduceLROnPlateau.on_test_batch_begin()
    • ReduceLROnPlateau.on_test_batch_end()
    • ReduceLROnPlateau.on_test_begin()
    • ReduceLROnPlateau.on_test_end()
    • ReduceLROnPlateau.on_train_batch_begin()
    • ReduceLROnPlateau.on_train_batch_end()
    • ReduceLROnPlateau.on_train_begin()
    • ReduceLROnPlateau.on_train_end()
    • ReduceLROnPlateau.set_model()
    • ReduceLROnPlateau.set_params()
• gradgraph.optimization.tf.constraints module
  • InBetween
    • InBetween.__call__()
    • InBetween.get_config()
    • InBetween.from_config()
    • InBetween.from_config()
    • InBetween.get_config()
• gradgraph.optimization.tf.interp module
  • batch_linear_interp_1d()
• gradgraph.optimization.tf.layers module
  • BasePDESystemLayer
    • BasePDESystemLayer.add_global_embedding()
    • BasePDESystemLayer.add_global_weight()
    • BasePDESystemLayer.add_local_embedding()
    • BasePDESystemLayer.add_local_weight()
    • BasePDESystemLayer.add_loss()
    • BasePDESystemLayer.add_metric()
    • BasePDESystemLayer.add_variable()
    • BasePDESystemLayer.add_weight()
    • BasePDESystemLayer.body()
    • BasePDESystemLayer.build()
    • BasePDESystemLayer.build_from_config()
    • BasePDESystemLayer.call()
    • BasePDESystemLayer.compute_dtype
    • BasePDESystemLayer.compute_mask()
    • BasePDESystemLayer.compute_output_shape()
    • BasePDESystemLayer.compute_output_spec()
    • BasePDESystemLayer.cond()
    • BasePDESystemLayer.count_params()
    • BasePDESystemLayer.dtype
    • BasePDESystemLayer.dtype_policy
    • BasePDESystemLayer.from_config()
    • BasePDESystemLayer.get_build_config()
    • BasePDESystemLayer.get_config()
    • BasePDESystemLayer.get_weights()
    • BasePDESystemLayer.global_non_trainable_weights
    • BasePDESystemLayer.global_trainable_weights
    • BasePDESystemLayer.global_weights
    • BasePDESystemLayer.input
    • BasePDESystemLayer.input_dtype
    • BasePDESystemLayer.input_spec
    • BasePDESystemLayer.load_own_variables()
    • BasePDESystemLayer.local_non_trainable_weights
    • BasePDESystemLayer.local_trainable_weights
    • BasePDESystemLayer.local_weights
    • BasePDESystemLayer.losses
    • BasePDESystemLayer.metrics
    • BasePDESystemLayer.metrics_variables
    • BasePDESystemLayer.non_trainable_variables
    • BasePDESystemLayer.non_trainable_weights
    • BasePDESystemLayer.output
    • BasePDESystemLayer.path
    • BasePDESystemLayer.quantization_mode
    • BasePDESystemLayer.quantize()
    • BasePDESystemLayer.quantized_build()
    • BasePDESystemLayer.quantized_call()
    • BasePDESystemLayer.rematerialized_call()
    • BasePDESystemLayer.save_own_variables()
    • BasePDESystemLayer.set_weights()
    • BasePDESystemLayer.stateless_call()
    • BasePDESystemLayer.supports_masking
    • BasePDESystemLayer.symbolic_call()
    • BasePDESystemLayer.trainable
    • BasePDESystemLayer.trainable_variables
    • BasePDESystemLayer.trainable_weights
    • BasePDESystemLayer.variable_dtype
    • BasePDESystemLayer.variables
    • BasePDESystemLayer.weights
  • Embedding
    • Embedding.build()
    • Embedding.add_loss()
    • Embedding.add_metric()
    • Embedding.add_variable()
    • Embedding.add_weight()
    • Embedding.build()
    • Embedding.build_from_config()
    • Embedding.call()
    • Embedding.compute_dtype
    • Embedding.compute_mask()
    • Embedding.compute_output_shape()
    • Embedding.compute_output_spec()
    • Embedding.count_params()
    • Embedding.dtype
    • Embedding.dtype_policy
    • Embedding.embeddings
    • Embedding.enable_lora()
    • Embedding.from_config()
    • Embedding.get_build_config()
    • Embedding.get_config()
    • Embedding.get_weights()
    • Embedding.input
    • Embedding.input_dtype
    • Embedding.input_spec
    • Embedding.load_own_variables()
    • Embedding.losses
    • Embedding.metrics
    • Embedding.metrics_variables
    • Embedding.non_trainable_variables
    • Embedding.non_trainable_weights
    • Embedding.output
    • Embedding.path
    • Embedding.quantization_mode
    • Embedding.quantize()
    • Embedding.quantized_build()
    • Embedding.quantized_call()
    • Embedding.rematerialized_call()
    • Embedding.save_own_variables()
    • Embedding.set_weights()
    • Embedding.stateless_call()
    • Embedding.supports_masking
    • Embedding.symbolic_call()
    • Embedding.trainable
    • Embedding.trainable_variables
    • Embedding.trainable_weights
    • Embedding.variable_dtype
    • Embedding.variables
    • Embedding.weights
• gradgraph.optimization.tf.trainer module
  • BasePDESystemTrainer
    • BasePDESystemTrainer.add_loss()
    • BasePDESystemTrainer.add_metric()
    • BasePDESystemTrainer.add_variable()
    • BasePDESystemTrainer.add_weight()
    • BasePDESystemTrainer.build()
    • BasePDESystemTrainer.build_from_config()
    • BasePDESystemTrainer.call()
    • BasePDESystemTrainer.compile()
    • BasePDESystemTrainer.compile_from_config()
    • BasePDESystemTrainer.compiled_loss()
    • BasePDESystemTrainer.compiled_metrics
    • BasePDESystemTrainer.compute_dtype
    • BasePDESystemTrainer.compute_loss()
    • BasePDESystemTrainer.compute_mask()
    • BasePDESystemTrainer.compute_metrics()
    • BasePDESystemTrainer.compute_output_shape()
    • BasePDESystemTrainer.compute_output_spec()
    • BasePDESystemTrainer.count_params()
    • BasePDESystemTrainer.distribute_reduction_method
    • BasePDESystemTrainer.distribute_strategy
    • BasePDESystemTrainer.dtype
    • BasePDESystemTrainer.dtype_policy
    • BasePDESystemTrainer.evaluate()
    • BasePDESystemTrainer.export()
    • BasePDESystemTrainer.fit()
    • BasePDESystemTrainer.from_config()
    • BasePDESystemTrainer.get_build_config()
    • BasePDESystemTrainer.get_compile_config()
    • BasePDESystemTrainer.get_config()
    • BasePDESystemTrainer.get_layer()
    • BasePDESystemTrainer.get_metrics_result()
    • BasePDESystemTrainer.get_state_tree()
    • BasePDESystemTrainer.get_weights()
    • BasePDESystemTrainer.global_non_trainable_weights
    • BasePDESystemTrainer.global_trainable_weights
    • BasePDESystemTrainer.global_weights
    • BasePDESystemTrainer.input
    • BasePDESystemTrainer.input_dtype
    • BasePDESystemTrainer.input_spec
    • BasePDESystemTrainer.jit_compile
    • BasePDESystemTrainer.layers
    • BasePDESystemTrainer.load_own_variables()
    • BasePDESystemTrainer.load_weights()
    • BasePDESystemTrainer.local_non_trainable_weights
    • BasePDESystemTrainer.local_trainable_weights
    • BasePDESystemTrainer.local_weights
    • BasePDESystemTrainer.loss()
    • BasePDESystemTrainer.losses
    • BasePDESystemTrainer.make_predict_function()
    • BasePDESystemTrainer.make_test_function()
    • BasePDESystemTrainer.make_train_function()
    • BasePDESystemTrainer.metrics
    • BasePDESystemTrainer.metrics_names
    • BasePDESystemTrainer.metrics_variables
    • BasePDESystemTrainer.non_trainable_variables
    • BasePDESystemTrainer.non_trainable_weights
    • BasePDESystemTrainer.output
    • BasePDESystemTrainer.path
    • BasePDESystemTrainer.predict()
    • BasePDESystemTrainer.predict_on_batch()
    • BasePDESystemTrainer.predict_step()
    • BasePDESystemTrainer.quantization_mode
    • BasePDESystemTrainer.quantize()
    • BasePDESystemTrainer.quantized_build()
    • BasePDESystemTrainer.quantized_call()
    • BasePDESystemTrainer.rematerialized_call()
    • BasePDESystemTrainer.reset_metrics()
    • BasePDESystemTrainer.run_eagerly
    • BasePDESystemTrainer.save()
    • BasePDESystemTrainer.save_own_variables()
    • BasePDESystemTrainer.save_weights()
    • BasePDESystemTrainer.set_state_tree()
    • BasePDESystemTrainer.set_weights()
    • BasePDESystemTrainer.stateless_call()
    • BasePDESystemTrainer.stateless_compute_loss()
    • BasePDESystemTrainer.summary()
    • BasePDESystemTrainer.supports_masking
    • BasePDESystemTrainer.symbolic_call()
    • BasePDESystemTrainer.test_on_batch()
    • BasePDESystemTrainer.test_step()
    • BasePDESystemTrainer.to_json()
    • BasePDESystemTrainer.train_on_batch()
    • BasePDESystemTrainer.train_step()
    • BasePDESystemTrainer.trainable
    • BasePDESystemTrainer.trainable_variables
    • BasePDESystemTrainer.trainable_weights
    • BasePDESystemTrainer.variable_dtype
    • BasePDESystemTrainer.variables
    • BasePDESystemTrainer.weights
• gradgraph.optimization.tf.utils module
  • initialize_tensorflow()
  • unscale_loss_for_distribution()