Commit b22bd4e3 authored by Geoffrey Yu's avatar Geoffrey Yu
Browse files

update config so that it will update multimer loss otherwise fape loss calculation crashes

parent beccaa4b
......@@ -163,6 +163,9 @@ def model_config(
for k,v in multimer_model_config_update.items():
c.model[k] = v
for k,v in multimer_model_config_update['loss'].items():
c.loss[k] = v
# TODO: Change max_msa_clusters and max_extra_msa to multimer feats within model
if re.fullmatch("^model_[1-5]_multimer(_v2)?$", name):
#c.model.input_embedder.num_msa = 252
......
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