Quellcode durchsuchen

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

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Xiaomeng Zhao vor 3 Monaten
Ursprung
Commit
fae9fb9820
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  1. 1 1
      mineru/model/mfr/unimernet/Unimernet.py

+ 1 - 1
mineru/model/mfr/unimernet/Unimernet.py

@@ -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)