hifi-gan.ipynb

{"nbformat":4,"nbformat_minor":0,"metadata":{"colab":{"provenance":[]},"kernelspec":{"name":"python3","display_name":"Python 3"},"language_info":{"name":"python"},"gpuClass":"premium","accelerator":"GPU"},"cells":[{"cell_type":"markdown","source":["## Check Device"],"metadata":{"id":"wt-x_mlFb-aU"}},{"cell_type":"code","execution_count":null,"metadata":{"id":"MWbKHDBFoszH","colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"status":"ok","timestamp":1671587331978,"user_tz":-540,"elapsed":512,"user":{"displayName":"김주의","userId":"10890239295475734502"}},"outputId":"41838035-d8bf-45d6-a72c-91f874a54ea0"},"outputs":[{"output_type":"stream","name":"stdout","text":["Wed Dec 21 01:48:50 2022       \n","+-----------------------------------------------------------------------------+\n","| NVIDIA-SMI 460.32.03    Driver Version: 460.32.03    CUDA Version: 11.2     |\n","|-------------------------------+----------------------+----------------------+\n","| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |\n","| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |\n","|                               |                      |               MIG M. |\n","|===============================+======================+======================|\n","|   0  Tesla T4            Off  | 00000000:00:04.0 Off |                    0 |\n","| N/A   50C    P0    26W /  70W |      0MiB / 15109MiB |      0%      Default |\n","|                               |                      |                  N/A |\n","+-------------------------------+----------------------+----------------------+\n","                                                                               \n","+-----------------------------------------------------------------------------+\n","| Processes:                                                                  |\n","|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |\n","|        ID   ID                                                   Usage      |\n","|=============================================================================|\n","|  No running processes found                                                 |\n","+-----------------------------------------------------------------------------+\n"]}],"source":["!nvidia-smi"]},{"cell_type":"markdown","source":["## Mount Google Drvie & OneDrive to Colab"],"metadata":{"id":"arM92cWLcEm1"}},{"cell_type":"code","source":["from google.colab import drive\n","drive.mount('/content/drive', force_remount=True)"],"metadata":{"id":"gNzK1J5wrUCy","colab":{"base_uri":"https://localhost:8080/"},"executionInfo":{"status":"ok","timestamp":1671587357940,"user_tz":-540,"elapsed":25969,"user":{"displayName":"김주의","userId":"10890239295475734502"}},"outputId":"0d0acb21-6cbd-45f2-de5e-e1ef79b64f69"},"execution_count":null,"outputs":[{"output_type":"stream","name":"stdout","text":["Mounted at /content/drive\n"]}]},{"cell_type":"code","source":["!python -m pip install --upgrade pip setuptools wheel\n","# https://colab.research.google.com/github/jakiya99/Torrent-to-Onedrive/blob/main/new_torrents_download_to_onedrive.ipynb#scrollTo=Fil_bHuAD8S3\n","# https://github.com/jakiyaa/Torrent-to-Onedrive/blob/main/full%20tutorial.md\n","# https://github.com/jakiyaa/rclone-authenticate\n","\n","!wget https://downloads.rclone.org/v1.60.1/rclone-v1.60.1-linux-amd64.deb\n","!apt install ./rclone-v1.60.1-linux-amd64.deb"],"metadata":{"id":"Qqb0ajplrUKo"},"execution_count":null,"outputs":[]},{"cell_type":"code","source":["!rclone config"],"metadata":{"id":"8iWyb_0zrUOe"},"execution_count":null,"outputs":[]},{"cell_type":"code","source":["!mkdir onedrive\n","# 사용자 shell이 종료되어도 작동되기 위해 nohup(no hang up), & 사용\n","!nohup rclone --vfs-cache-mode writes mount onedrive: /content/onedrive &"],"metadata":{"id":"tTQ5-gPILIXv"},"execution_count":null,"outputs":[]},{"cell_type":"code","source":["!ls ./onedrive"],"metadata":{"id":"dZlK42dILKU9"},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":["## Install TTS package"],"metadata":{"id":"Et_AtFJycKXz"}},{"cell_type":"code","source":["!pip install TTS==0.10.0"],"metadata":{"id":"2U59VO-truGS"},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":["## Train HiFi-GAN"],"metadata":{"id":"h8e3OZC1b4gs"}},{"cell_type":"code","source":["import os\n","\n","from trainer import Trainer, TrainerArgs\n","\n","from TTS.utils.audio import AudioProcessor\n","from TTS.vocoder.configs import HifiganConfig\n","from TTS.tts.configs.shared_configs import BaseAudioConfig\n","from TTS.vocoder.datasets.preprocess import load_wav_data\n","from TTS.vocoder.models.gan import GAN\n","\n","output_path = '/content/drive/MyDrive/glow-tts-dialect/model/hifigan/'\n","\n","# define audio config\n","audio_config = BaseAudioConfig(\n","    sample_rate=22050,\n","    resample=True,\n",")\n","\n","config = HifiganConfig(\n","    batch_size=64,\n","    eval_batch_size=16,\n","    num_loader_workers=4,\n","    num_eval_loader_workers=4,\n","    run_eval=True,\n","    test_delay_epochs=5,\n","    epochs=1000,\n","    seq_len=8192,\n","    pad_short=2000,\n","    use_noise_augment=True,\n","    print_step=10,\n","    print_eval=False,\n","    mixed_precision=False,\n","    lr_gen=1e-4,\n","    lr_disc=1e-4,\n","    data_path=\"/content/onedrive/jss-dataset\",\n","    output_path=output_path,\n","    audio=audio_config,\n",")\n","\n","# init the audio processor\n","ap = AudioProcessor.init_from_config(config)\n","\n","# load training samples\n","eval_samples, train_samples = load_wav_data(config.data_path, config.eval_split_size)\n","\n","# init the model\n","model = GAN(config, ap)\n","\n","# init the trainer and 🚀\n","trainer = Trainer(\n","    TrainerArgs(), config, output_path, model=model, train_samples=train_samples, eval_samples=eval_samples\n",")\n","\n","trainer.fit()"],"metadata":{"id":"N1bZHgPjrUQh"},"execution_count":null,"outputs":[]},{"cell_type":"code","source":[],"metadata":{"id":"Hd2ZdVWmUfCC"},"execution_count":null,"outputs":[]}]}