Skip to content

Commit

Permalink
set default value for kappa switch
Browse files Browse the repository at this point in the history
  • Loading branch information
RasmusOrsoe committed Sep 27, 2024
1 parent a68df6e commit d52e583
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/graphnet/training/loss_functions.py
Original file line number Diff line number Diff line change
Expand Up @@ -307,7 +307,7 @@ def log_cmk_approx(

@classmethod
def log_cmk(
cls, m: int, kappa: Tensor, kappa_switch: float = 100.0
cls, m: int, kappa: Tensor, kappa_switch: float = 600.0
) -> Tensor: # pylint: disable=invalid-name
"""Calculate $log C_{m}(k)$ term in von Mises-Fisher loss.
Expand Down

0 comments on commit d52e583

Please sign in to comment.