Skip to content

GitLab

"lib/llm/src/git@developer.sourcefind.cn:OpenDAS/dynamo.git" did not exist on "97bf8184023c0a47e9200526bed301a9c92971be"
Commit 833c100b authored by Geoffrey Yu's avatar Geoffrey Yu
Browse files

put multimer_model_config_update inside a 'model' key in the config dict

parent b8d00692
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
openfold/config.py
View file @ 833c100b
...@@ -160,7 +160,7 @@ def model_config( ...@@ -160,7 +160,7 @@ def model_config(
c.loss.masked_msa.num_classes = 22 c.loss.masked_msa.num_classes = 22
c.data.common.max_recycling_iters = 20 c.data.common.max_recycling_iters = 20
for k,v in multimer_model_config_update.items(): for k,v in multimer_model_config_update['model'].items():
c.model[k] = v c.model[k] = v
for k,v in multimer_model_config_update['loss'].items(): for k,v in multimer_model_config_update['loss'].items():
...@@ -683,7 +683,7 @@ config = mlc.ConfigDict( ...@@ -683,7 +683,7 @@ config = mlc.ConfigDict(
) )
multimer_model_config_update = { multimer_model_config_update = {
"input_embedder": { 'model':{"input_embedder": {
"tf_dim": 21, "tf_dim": 21,
"msa_dim": 49, "msa_dim": 49,
#"num_msa": 508, #"num_msa": 508,
...@@ -824,7 +824,7 @@ multimer_model_config_update = { ...@@ -824,7 +824,7 @@ multimer_model_config_update = {
"c_s": c_s, "c_s": c_s,
"c_out": 37, "c_out": 37,
}, },
}, }},
"loss": { "loss": {
"distogram": { "distogram": {
"min_bin": 2.3125, "min_bin": 2.3125,
......
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