Skip to content

Commit

Permalink
Dorado Correct: Log the batch size to info.
Browse files Browse the repository at this point in the history
  • Loading branch information
John Stone committed Jul 30, 2024
1 parent 9b5ba45 commit bc15a26
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion dorado/read_pipeline/CorrectionNode.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -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));
}
Expand Down

0 comments on commit bc15a26

Please sign in to comment.