changed the way to determine the batch size for loading data into GPU…

… added INFO log
hzi-bifo · Apr 19, 2021 · 3942daa · 3942daa
1 parent 5815890
commit 3942daa
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/detect.py b/detect.py
@@ -510,6 +510,15 @@ def detect(self):
             self.read_collate_fn = partial(
                 unlabeled_read_collate_fn, max_len=self.len, pack_seq=self.pack_seq)
 
+        self.logger.info('Choose batch size: {}{}{}{} based on the given GPU RAM size {}GB and max read length {}'.format(
+            colors.BOLD,
+            colors.OKCYAN,
+            self.batch_size,
+            colors.ENDC,
+            self.args.memory,
+            self.len
+        ))
+
         if self.chunk_size is None:
             self.run()
         else: