项目作者: Doyosae

项目描述 :
Implementation of TC-Resnet
高级语言: Python
项目地址: git://github.com/Doyosae/Temporal-Convolution-Resnet.git
创建时间: 2021-07-23T19:37:38Z
项目社区:https://github.com/Doyosae/Temporal-Convolution-Resnet

开源协议:

下载


Temporal-Convolution Resnet (PyTorch)

TC-Resnet is speech deep neural network for KWS task (Hyperconnect)
Original repository is tensorflow version code
So, This repository gives you pytorch version code

Requirements

  1. pytorch == 1.8.0
  2. torchaudio == 0.8.0
  3. or
  4. pytorch == 1.9.0
  5. torchaudio == 0.9.0
  6. tqdm
  7. einops
  8. numba == 0.48
  9. librosa == 0.7.2

Setting enviorments

  1. Install conda and setting enviorment
    1. bash conda_install.sh
    2. conda create -n tc-resnet python=3.8
  2. Install some packages
    1. conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
    2. pip install numba==0.48
    3. pip install librosa==0.7.2
    4. pip install tqdm
    5. pip intsall einops
  3. Or, You can select one-shot option
    1. conda env create -f pakages.yml

    Preparing dataset

    1. bash download.sh

    Training model

    1. python model_train.py