From bc15a264fdda254ea0d691800dcf891ed7888a00 Mon Sep 17 00:00:00 2001 From: John Stone Date: Thu, 25 Jul 2024 10:32:43 +0100 Subject: [PATCH] Dorado Correct: Log the batch size to info. --- dorado/read_pipeline/CorrectionNode.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/dorado/read_pipeline/CorrectionNode.cpp b/dorado/read_pipeline/CorrectionNode.cpp index 231ba1572..3890dc6bd 100644 --- a/dorado/read_pipeline/CorrectionNode.cpp +++ b/dorado/read_pipeline/CorrectionNode.cpp @@ -461,7 +461,8 @@ CorrectionNode::CorrectionNode(const std::string& fastq, throw std::runtime_error("Insufficient memory to run inference on " + dev); } } - spdlog::debug("Using batch size {} on device {}", device_batch_size, dev); + spdlog::info("> Using batch size {} on device {} in inference thread {}.", + device_batch_size, dev, i); m_infer_threads.push_back( std::thread(&CorrectionNode::infer_fn, this, dev, (int)d, device_batch_size)); }