Skip to content
Snippets Groups Projects

Document batch prediction arg

Merged Yoann Schneider requested to merge document-batch-size into main
1 file
+ 1
0
Compare changes
  • Side-by-side
  • Inline
+ 1
0
@@ -23,6 +23,7 @@ Use the `teklia-dan predict` command to apply a trained DAN model on an image.
| `--line-separators` | List of line separators. | `list` | `["\n"]` |
| `--threshold-method` | Method to use for attention mask thresholding. Should be in `["otsu", "simple"]`. | `str` | `"otsu"` |
| `--threshold-value ` | Threshold to use for the "simple" thresholding method. | `int` | `0` |
| `--batch-size ` | Size of the batches for prediction. | `int` | `1` |
## Examples
Loading