Skip to content

GitLab

Unverified Commit 71735d5d authored by double-vin's avatar double-vin Committed by GitHub
Browse files

added relaxation argument (#146) 


Co-authored-by: default avatarzhuww <zhuww@sugon.com>
parent 4dc719f7
Show whitespace changes
Inline Side-by-side
Showing with 31 additions and 26 deletions
inference.py
View file @ 71735d5d
...@@ -306,6 +306,7 @@ def inference_multimer_model(args): ...@@ -306,6 +306,7 @@ def inference_multimer_model(args):
with open(unrelaxed_output_path, 'w') as f: with open(unrelaxed_output_path, 'w') as f:
f.write(protein.to_pdb(unrelaxed_protein)) f.write(protein.to_pdb(unrelaxed_protein))
if(args.relaxation):
amber_relaxer = relax.AmberRelaxation( amber_relaxer = relax.AmberRelaxation(
use_gpu=True, use_gpu=True,
**config.relax, **config.relax,
...@@ -445,6 +446,7 @@ def inference_monomer_model(args): ...@@ -445,6 +446,7 @@ def inference_monomer_model(args):
with open(unrelaxed_output_path, 'w') as f: with open(unrelaxed_output_path, 'w') as f:
f.write(protein.to_pdb(unrelaxed_protein)) f.write(protein.to_pdb(unrelaxed_protein))
if(args.relaxation):
amber_relaxer = relax.AmberRelaxation( amber_relaxer = relax.AmberRelaxation(
use_gpu=True, use_gpu=True,
**config.relax, **config.relax,
...@@ -494,6 +496,9 @@ if __name__ == "__main__": ...@@ -494,6 +496,9 @@ if __name__ == "__main__":
help="""Path to model parameters. If None, parameters are selected help="""Path to model parameters. If None, parameters are selected
automatically according to the model name from automatically according to the model name from
./data/params""") ./data/params""")
parser.add_argument(
"--relaxation", action="store_false", default=False,
)
parser.add_argument("--cpus", parser.add_argument("--cpus",
type=int, type=int,
default=12, default=12,
......
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