Yabin Li
2023-08-16 52d554bde25ac3c0605bb0fa2c1c8a703c37f3b6
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