Skip to content

GitLab

".github/vscode:/vscode.git/clone" did not exist on "c80490a95e935c783686401287301a550cc2f5f2"
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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
alphafold/model/folding_multimer.py
View file @ 5352e298
...@@ -789,7 +789,7 @@ def backbone_loss(gt_rigid: geometry.Rigid3Array, ...@@ -789,7 +789,7 @@ def backbone_loss(gt_rigid: geometry.Rigid3Array,
loss_fn = functools.partial( loss_fn = functools.partial(
all_atom_multimer.frame_aligned_point_error, all_atom_multimer.frame_aligned_point_error,
l1_clamp_distance=config.atom_clamp_distance, 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)) loss_fn = jax.vmap(loss_fn, (0, None, None, 0, None, None, None))
fape = loss_fn(target_rigid, gt_rigid, gt_frames_mask, 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