Skip to content

Commit dba9bba

Browse files
mihowDebian
authored andcommitted
Increase batch size sent to ML backend
1 parent 9554171 commit dba9bba

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ami/jobs/models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -418,7 +418,7 @@ def run(self):
418418
total_detections = 0
419419
total_classifications = 0
420420

421-
CHUNK_SIZE = 2 # Keep it low to see more progress updates
421+
CHUNK_SIZE = 4 # Keep it low to see more progress updates
422422
chunks = [images[i : i + CHUNK_SIZE] for i in range(0, image_count, CHUNK_SIZE)] # noqa
423423

424424
for i, chunk in enumerate(chunks):

0 commit comments

Comments
 (0)