Yabin Li
2023-09-26 b1b6381cd5ad135f4d5aab7a7ef93b6bfc0e7bb7
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