Fix incorrect length_scale argument

PiperOrigin-RevId: 474834335 Change-Id: I39693e87f05c6306437e276b5539cf93cbfca0dc
google-deepmind · Sep 16, 2022 · 5352e29 · 5352e29
1 parent 54c7358
commit 5352e29
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/alphafold/model/folding_multimer.py b/alphafold/model/folding_multimer.py
@@ -789,7 +789,7 @@ def backbone_loss(gt_rigid: geometry.Rigid3Array,
   loss_fn = functools.partial(
       all_atom_multimer.frame_aligned_point_error,
       l1_clamp_distance=config.atom_clamp_distance,
-      loss_unit_distance=config.loss_unit_distance)
+      length_scale=config.loss_unit_distance)
 
   loss_fn = jax.vmap(loss_fn, (0, None, None, 0, None, None, None))
   fape = loss_fn(target_rigid, gt_rigid, gt_frames_mask,