"lib/engines/python/src/lib.rs" did not exist on "4f7f4b40186f6f32b3ce76b8b26c0c6056e2a373"
Commit dd8e44b3 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Add missing config values

parent ab0c6977
...@@ -19,7 +19,8 @@ c_m = mlc.FieldReference(256) ...@@ -19,7 +19,8 @@ c_m = mlc.FieldReference(256)
c_t = mlc.FieldReference(64) c_t = mlc.FieldReference(64)
c_e = mlc.FieldReference(64) c_e = mlc.FieldReference(64)
c_s = mlc.FieldReference(384) c_s = mlc.FieldReference(384)
chunk_size = mlc.FieldReference(4) blocks_per_ckpt = mlc.FieldReference(1)
chunk_size = mlc.FieldReference(4)#1280)
aux_distogram_bins = mlc.FieldReference(64) aux_distogram_bins = mlc.FieldReference(64)
config = mlc.ConfigDict({ config = mlc.ConfigDict({
...@@ -71,7 +72,7 @@ config = mlc.ConfigDict({ ...@@ -71,7 +72,7 @@ config = mlc.ConfigDict({
"no_heads": 4, "no_heads": 4,
"pair_transition_n": 2, "pair_transition_n": 2,
"dropout_rate": 0.25, "dropout_rate": 0.25,
"blocks_per_ckpt": None, "blocks_per_ckpt": blocks_per_ckpt,
"chunk_size": chunk_size, "chunk_size": chunk_size,
}, },
"template_pointwise_attention": { "template_pointwise_attention": {
...@@ -105,7 +106,7 @@ config = mlc.ConfigDict({ ...@@ -105,7 +106,7 @@ config = mlc.ConfigDict({
"transition_n": 4, "transition_n": 4,
"msa_dropout": 0.15, "msa_dropout": 0.15,
"pair_dropout": 0.25, "pair_dropout": 0.25,
"blocks_per_ckpt": None, "blocks_per_ckpt": blocks_per_ckpt,
"chunk_size": chunk_size, "chunk_size": chunk_size,
"inf": 1e9, "inf": 1e9,
"eps": 1e-10, "eps": 1e-10,
...@@ -126,7 +127,7 @@ config = mlc.ConfigDict({ ...@@ -126,7 +127,7 @@ config = mlc.ConfigDict({
"transition_n": 4, "transition_n": 4,
"msa_dropout": 0.15, "msa_dropout": 0.15,
"pair_dropout": 0.25, "pair_dropout": 0.25,
"blocks_per_ckpt": None, "blocks_per_ckpt": blocks_per_ckpt,
"chunk_size": chunk_size, "chunk_size": chunk_size,
"inf": 1e9, "inf": 1e9,
"eps": 1e-10, "eps": 1e-10,
...@@ -226,6 +227,8 @@ config = mlc.ConfigDict({ ...@@ -226,6 +227,8 @@ config = mlc.ConfigDict({
"weight": 1.0, "weight": 1.0,
}, },
"violation": { "violation": {
"violation_tolerance_factor": 12.0,
"clash_overlap_tolerance": 1.5,
"eps": 1e-6, "eps": 1e-6,
"weight": 0., "weight": 0.,
}, },
......
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