diff --git a/docs/usage/predict.md b/docs/usage/predict.md index 94a241dcaa0bdea6f6e65f3778c4e9b2825e5b3d..294e67aafd54b7a5750d4d99d7df0a6b4cf1a143 100644 --- a/docs/usage/predict.md +++ b/docs/usage/predict.md @@ -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