Commit 5352e298 authored by Hamish Tomlinson's avatar Hamish Tomlinson Committed by Copybara-Service
Browse files

Fix incorrect length_scale argument

PiperOrigin-RevId: 474834335
Change-Id: I39693e87f05c6306437e276b5539cf93cbfca0dc
parent 54c73580
......@@ -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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment