"...git@developer.sourcefind.cn:jerrrrry/infinicore.git" did not exist on "0450fb1eb5c7ea608dd610632cfaeb4b825e683b"
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