Unverified Commit fae9fb98 authored by Xiaomeng Zhao's avatar Xiaomeng Zhao Committed by GitHub
Browse files

Update mineru/model/mfr/unimernet/Unimernet.py


Co-authored-by: default avatarCopilot <175728472+Copilot@users.noreply.github.com>
parent 58268df3
......@@ -105,7 +105,7 @@ class UnimernetModel(object):
# Create dataset with sorted images
dataset = MathDataset(sorted_images, transform=self.model.transform)
# 如果batch_size> len(sorted_images),则设置为不超过len(sorted_images)的2的阶乘
# 如果batch_size> len(sorted_images),则设置为不超过len(sorted_images)的2的
batch_size = min(batch_size, 2 ** (len(sorted_images).bit_length() - 1))
dataloader = DataLoader(dataset, batch_size=batch_size, num_workers=0)
......
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