deepmd.tf.loss.loss
Module Contents
Classes
The abstract class for the loss function. |
- class deepmd.tf.loss.loss.Loss[source]
The abstract class for the loss function.
- abstract build(learning_rate: deepmd.tf.env.tf.Tensor, natoms: deepmd.tf.env.tf.Tensor, model_dict: Dict[str, deepmd.tf.env.tf.Tensor], label_dict: Dict[str, deepmd.tf.env.tf.Tensor], suffix: str) Tuple[deepmd.tf.env.tf.Tensor, Dict[str, deepmd.tf.env.tf.Tensor]] [source]
Build the loss function graph.
- Parameters:
- Returns: