Add changes from pre-commit hook
This commit is contained in:
parent
2e5efe5e15
commit
9252cef138
|
@ -3,9 +3,8 @@
|
||||||
|
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
from google.protobuf import text_format
|
|
||||||
|
|
||||||
import tensorflow.compat.v1 as tfv1
|
import tensorflow.compat.v1 as tfv1
|
||||||
|
from google.protobuf import text_format
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
|
|
|
@ -11,7 +11,6 @@ import pandas
|
||||||
import progressbar
|
import progressbar
|
||||||
from coqui_stt_training.util.downloader import maybe_download
|
from coqui_stt_training.util.downloader import maybe_download
|
||||||
from sox import Transformer
|
from sox import Transformer
|
||||||
|
|
||||||
from tensorflow.python.platform import gfile
|
from tensorflow.python.platform import gfile
|
||||||
|
|
||||||
SAMPLE_RATE = 16000
|
SAMPLE_RATE = 16000
|
||||||
|
|
|
@ -10,7 +10,6 @@ import pandas
|
||||||
from coqui_stt_training.util.downloader import maybe_download
|
from coqui_stt_training.util.downloader import maybe_download
|
||||||
from coqui_stt_training.util.stm import parse_stm_file
|
from coqui_stt_training.util.stm import parse_stm_file
|
||||||
from sox import Transformer
|
from sox import Transformer
|
||||||
|
|
||||||
from tensorflow.python.platform import gfile
|
from tensorflow.python.platform import gfile
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -14,7 +14,6 @@ from os import makedirs, path
|
||||||
import pandas
|
import pandas
|
||||||
from bs4 import BeautifulSoup
|
from bs4 import BeautifulSoup
|
||||||
from coqui_stt_training.util.downloader import maybe_download
|
from coqui_stt_training.util.downloader import maybe_download
|
||||||
|
|
||||||
from tensorflow.python.platform import gfile
|
from tensorflow.python.platform import gfile
|
||||||
|
|
||||||
"""The number of jobs to run in parallel"""
|
"""The number of jobs to run in parallel"""
|
||||||
|
|
|
@ -6,6 +6,7 @@ import sys
|
||||||
|
|
||||||
import absl.app
|
import absl.app
|
||||||
import optuna
|
import optuna
|
||||||
|
import tensorflow.compat.v1 as tfv1
|
||||||
from coqui_stt_ctcdecoder import Scorer
|
from coqui_stt_ctcdecoder import Scorer
|
||||||
from coqui_stt_training.evaluate import evaluate
|
from coqui_stt_training.evaluate import evaluate
|
||||||
from coqui_stt_training.train import create_model
|
from coqui_stt_training.train import create_model
|
||||||
|
@ -14,8 +15,6 @@ from coqui_stt_training.util.evaluate_tools import wer_cer_batch
|
||||||
from coqui_stt_training.util.flags import FLAGS, create_flags
|
from coqui_stt_training.util.flags import FLAGS, create_flags
|
||||||
from coqui_stt_training.util.logging import log_error
|
from coqui_stt_training.util.logging import log_error
|
||||||
|
|
||||||
import tensorflow.compat.v1 as tfv1
|
|
||||||
|
|
||||||
|
|
||||||
def character_based():
|
def character_based():
|
||||||
is_character_based = False
|
is_character_based = False
|
||||||
|
|
|
@ -7,11 +7,11 @@ import sys
|
||||||
from multiprocessing import cpu_count
|
from multiprocessing import cpu_count
|
||||||
|
|
||||||
import progressbar
|
import progressbar
|
||||||
|
import tensorflow.compat.v1 as tfv1
|
||||||
from coqui_stt_ctcdecoder import Scorer, ctc_beam_search_decoder_batch
|
from coqui_stt_ctcdecoder import Scorer, ctc_beam_search_decoder_batch
|
||||||
from six.moves import zip
|
from six.moves import zip
|
||||||
|
|
||||||
import tensorflow as tf
|
import tensorflow as tf
|
||||||
import tensorflow.compat.v1 as tfv1
|
|
||||||
|
|
||||||
from .util.augmentations import NormalizeSampleRate
|
from .util.augmentations import NormalizeSampleRate
|
||||||
from .util.checkpoints import load_graph_for_evaluation
|
from .util.checkpoints import load_graph_for_evaluation
|
||||||
|
|
|
@ -17,9 +17,9 @@ import time
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import progressbar
|
import progressbar
|
||||||
|
import tensorflow.compat.v1 as tfv1
|
||||||
|
|
||||||
import tensorflow as tf
|
import tensorflow as tf
|
||||||
import tensorflow.compat.v1 as tfv1
|
|
||||||
|
|
||||||
tfv1.logging.set_verbosity(
|
tfv1.logging.set_verbosity(
|
||||||
{
|
{
|
||||||
|
|
|
@ -1,8 +1,9 @@
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
import tensorflow as tf
|
|
||||||
import tensorflow.compat.v1 as tfv1
|
import tensorflow.compat.v1 as tfv1
|
||||||
|
|
||||||
|
import tensorflow as tf
|
||||||
|
|
||||||
from .config import Config, log_error, log_info, log_warn
|
from .config import Config, log_error, log_info, log_warn
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -6,13 +6,12 @@ from dataclasses import asdict, dataclass, field
|
||||||
from typing import List
|
from typing import List
|
||||||
|
|
||||||
import progressbar
|
import progressbar
|
||||||
|
import tensorflow.compat.v1 as tfv1
|
||||||
from attrdict import AttrDict
|
from attrdict import AttrDict
|
||||||
from coqpit import MISSING, Coqpit, check_argument
|
from coqpit import MISSING, Coqpit, check_argument
|
||||||
from coqui_stt_ctcdecoder import Alphabet, UTF8Alphabet
|
from coqui_stt_ctcdecoder import Alphabet, UTF8Alphabet
|
||||||
from xdg import BaseDirectory as xdg
|
from xdg import BaseDirectory as xdg
|
||||||
|
|
||||||
import tensorflow.compat.v1 as tfv1
|
|
||||||
|
|
||||||
from .augmentations import NormalizeSampleRate, parse_augmentations
|
from .augmentations import NormalizeSampleRate, parse_augmentations
|
||||||
from .gpu import get_available_gpus
|
from .gpu import get_available_gpus
|
||||||
from .helpers import parse_file_size
|
from .helpers import parse_file_size
|
||||||
|
|
|
@ -5,9 +5,9 @@ from collections import Counter
|
||||||
from functools import partial
|
from functools import partial
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
from tensorflow.python.ops import gen_audio_ops as contrib_audio
|
||||||
|
|
||||||
import tensorflow as tf
|
import tensorflow as tf
|
||||||
from tensorflow.python.ops import gen_audio_ops as contrib_audio
|
|
||||||
|
|
||||||
from .audio import DEFAULT_FORMAT, pcm_to_np, read_frames_from_file, vad_split
|
from .audio import DEFAULT_FORMAT, pcm_to_np, read_frames_from_file, vad_split
|
||||||
from .augmentations import apply_graph_augmentations, apply_sample_augmentations
|
from .augmentations import apply_graph_augmentations, apply_sample_augmentations
|
||||||
|
|
|
@ -7,9 +7,10 @@ import os
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
|
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
|
||||||
import tensorflow as tf
|
|
||||||
import tensorflow.compat.v1.logging as tflogging
|
import tensorflow.compat.v1.logging as tflogging
|
||||||
|
|
||||||
|
import tensorflow as tf
|
||||||
|
|
||||||
tflogging.set_verbosity(tflogging.ERROR)
|
tflogging.set_verbosity(tflogging.ERROR)
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue