tensorclouds.loss.losses

Attributes

CA_INDEX

Classes

InternalVectorLoss

Functions

vector_cloud_matching_loss(...)

Module Contents

tensorclouds.loss.losses.CA_INDEX = 1

tensorclouds.loss.losses.vector_cloud_matching_loss(self, rng_key, model_output: tensorclouds.nn.utils.ModelOutput, _: moleculib.protein.datum.ProteinDatum) → Tuple[tensorclouds.nn.utils.ModelOutput, jax.Array, Dict[str, float]]

class tensorclouds.loss.losses.InternalVectorLoss(weight=1.0, start_step=0, norm_only=False)

norm_only = False