Commit 267b1bfd authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Add warning to run script

parent a3a27143
...@@ -182,7 +182,7 @@ if __name__ == "__main__": ...@@ -182,7 +182,7 @@ if __name__ == "__main__":
) )
parser.add_argument( parser.add_argument(
"--cpus", type=int, default=4, "--cpus", type=int, default=4,
help="""Number of CPUs to use to run alignment tools""" help="""Number of CPUs with which to run alignment tools"""
) )
parser.add_argument( parser.add_argument(
'--preset', type=str, default='full_dbs', '--preset', type=str, default='full_dbs',
...@@ -200,6 +200,12 @@ if __name__ == "__main__": ...@@ -200,6 +200,12 @@ if __name__ == "__main__":
"params_" + args.model_name + ".npz" "params_" + args.model_name + ".npz"
) )
if(args.model_device == "cpu" and torch.cuda.is_available()):
logging.warning(
"""The model is being run on CPU. Consider specifying
--model_device for better performance"""
)
if(args.bfd_database_path is None and if(args.bfd_database_path is None and
args.small_bfd_database_path is None): args.small_bfd_database_path is None):
raise ValueError( raise ValueError(
......
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