shixian.shi
2023-10-10 8a0930d682fe3206e0b41c694fc03d7d10c7eed2
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