Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
OpenFold
Commits
204ed191
"lib/llm/git@developer.sourcefind.cn:OpenDAS/dynamo.git" did not exist on "e1ae9aa09bcc1f81a986aa65bf9bb2d66e1cffc4"
Commit
204ed191
authored
Feb 07, 2024
by
Jennifer
Browse files
create separate test configs for multimer and monomer
parent
de12c0ea
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
28 additions
and
1 deletion
+28
-1
tests/config.py
tests/config.py
+28
-1
No files found.
tests/config.py
View file @
204ed191
import
ml_collections
as
mlc
import
ml_collections
as
mlc
consts
=
mlc
.
ConfigDict
(
monomer_consts
=
mlc
.
ConfigDict
(
{
"model"
:
"model_1_ptm"
,
# monomer:model_1_ptm, multimer: model_1_multimer_v3
"is_multimer"
:
False
,
# monomer: False, multimer: True
"chunk_size"
:
4
,
"batch_size"
:
2
,
"n_res"
:
22
,
"n_seq"
:
13
,
"n_templ"
:
3
,
"n_extra"
:
17
,
"n_heads_extra_msa"
:
8
,
"eps"
:
5e-4
,
# For compatibility with DeepMind's pretrained weights, it's easiest for
# everyone if these take their real values.
"c_m"
:
256
,
"c_z"
:
128
,
"c_s"
:
384
,
"c_t"
:
64
,
"c_e"
:
64
,
"msa_logits"
:
23
,
# monomer: 23, multimer: 22
"template_mmcif_dir"
:
None
# Set for test_multimer_datamodule
}
)
multimer_consts
=
mlc
.
ConfigDict
(
{
{
"model"
:
"model_1_multimer_v3"
,
# monomer:model_1_ptm, multimer: model_1_multimer_v3
"model"
:
"model_1_multimer_v3"
,
# monomer:model_1_ptm, multimer: model_1_multimer_v3
"is_multimer"
:
True
,
# monomer: False, multimer: True
"is_multimer"
:
True
,
# monomer: False, multimer: True
...
@@ -24,6 +49,8 @@ consts = mlc.ConfigDict(
...
@@ -24,6 +49,8 @@ consts = mlc.ConfigDict(
}
}
)
)
consts
=
monomer_consts
config
=
mlc
.
ConfigDict
(
config
=
mlc
.
ConfigDict
(
{
{
"data"
:
{
"data"
:
{
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment