Make ESMFold predictor more verbose

BioinfoMachineLearning · Dec 21, 2024 · 330f02b · 330f02b
1 parent 462d160
commit 330f02b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/posebench/data/components/esmfold_batch_structure_prediction.py b/posebench/data/components/esmfold_batch_structure_prediction.py
@@ -202,6 +202,9 @@ def run(args: argparse.Namespace):
     num_completed = 0
     num_sequences = len(all_sequences)
     for headers, sequences in batched_sequences:
+        logger.info(
+            f"Predicting structures for {len(sequences)} sequences of {headers} with total length {sum(len(seq) for seq in sequences)}"
+        )
         start = timer()
         try:
             output = model.infer(sequences, num_recycles=args.num_recycles)