qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-04-14 19:31:57 +02:00
Code Issues Projects Releases Wiki Activity

predictor: disable rebatching (until we have flexible batch sizes)

Browse source
This commit is contained in:
Robert Sachunsky 2026-04-13 01:14:49 +02:00
parent 219954d15b
commit f44c39667e
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/eynollah/predictor.py
View file

@ -111,7 +111,7 @@ class Predictor(mp.context.SpawnProcess):
"binarization": 4, "binarization": 4,
"enhancement": 4, "enhancement": 4,
"reading_order": 4, "reading_order": 4,
# medium size (672x672)... # medium size (672x672x3)...
"textline": 2, "textline": 2,
# large models... # large models...
"table": 1, "table": 1,
@ -119,6 +119,7 @@ class Predictor(mp.context.SpawnProcess):
"region_fl_np": 1, "region_fl_np": 1,
"region_fl": 1, "region_fl": 1,
}.get(self.name, 1) }.get(self.name, 1)
REBATCH_SIZE = 1 # save VRAM; FIXME: re-enable w/ runtime parameter
if not len(shared_data): if not len(shared_data):
#self.logger.debug("getting '%d' output shape of model '%s'", jobid, self.name) #self.logger.debug("getting '%d' output shape of model '%s'", jobid, self.name)
result = self.model.output_shape result = self.model.output_shape