| | |
|---|
| | | # -*- encoding: utf-8 -*- |
|---|
| | | #!/usr/bin/env python3 |
|---|
| | | # Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights Reserved. |
|---|
| | | # MIT License (https://opensource.org/licenses/MIT) |
|---|
| | | |
|---|
| | | import os |
|---|
| | | from funasr.tasks.punctuation import PunctuationTask |
|---|
| | | |
|---|
| | |
|---|
| | | else: |
|---|
| | | args.distributed = False |
|---|
| | | |
|---|
| | | if args.dataset_type == "small": |
|---|
| | | if args.batch_size is not None: |
|---|
| | | args.batch_size = args.batch_size * args.ngpu * args.num_worker_count |
|---|
| | | if args.batch_bins is not None: |
|---|
| | | args.batch_bins = args.batch_bins * args.ngpu * args.num_worker_count |
|---|
| | | |
|---|
| | | main(args=args) |
|---|
