Binbin Gu
2023-09-22 c1d01605bf5c4dc383f4c397ae4f566ce91b214a
funasr/bin/diar_inference_launch.py
@@ -453,11 +453,17 @@
        help="The batch size for inference",
    )
    group.add_argument(
        "--diar_smooth_size",
        "--smooth_size",
        type=int,
        default=121,
        help="The smoothing size for post-processing"
    )
    group.add_argument(
        "--dur_threshold",
        type=int,
        default=10,
        help="The threshold of minimum duration"
    )
    return parser