Skip to content

GitLab

"tools/git@developer.sourcefind.cn:OpenDAS/openpcdet.git" did not exist on "dd2edf9da0944238d3711f8145cc5b86c65f5a79"
Commit 376c60c7 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Rename run script parameter

parent a3c2ae51
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
run_pretrained_alphafold.py
View file @ 376c60c7
...@@ -42,8 +42,6 @@ from openfold.utils.tensor_utils import ( ...@@ -42,8 +42,6 @@ from openfold.utils.tensor_utils import (
tensor_tree_map, tensor_tree_map,
) )
FEAT_PATH = "tests/test_data/sample_feats.pickle"
MAX_TEMPLATE_HITS = 20 MAX_TEMPLATE_HITS = 20
def main(args): def main(args):
...@@ -81,7 +79,7 @@ def main(args): ...@@ -81,7 +79,7 @@ def main(args):
) )
output_dir_base = args.output_dir output_dir_base = args.output_dir
random_seed = args.random_seed random_seed = args.data_random_seed
if random_seed is None: if random_seed is None:
random_seed = random.randrange(sys.maxsize) random_seed = random.randrange(sys.maxsize)
config.data.eval.num_ensemble = num_ensemble config.data.eval.num_ensemble = num_ensemble
...@@ -237,7 +235,7 @@ if __name__ == "__main__": ...@@ -237,7 +235,7 @@ if __name__ == "__main__":
choices=('reduced_dbs', 'full_dbs') choices=('reduced_dbs', 'full_dbs')
) )
parser.add_argument( parser.add_argument(
'--random_seed', type=str, default=None '--data_random_seed', type=str, default=None
) )
args = parser.parse_args() args = parser.parse_args()
......
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