diff --git a/TrainNemo/.gitignore b/TrainNemo/.gitignore new file mode 100644 index 0000000..763513e --- /dev/null +++ b/TrainNemo/.gitignore @@ -0,0 +1 @@ +.ipynb_checkpoints diff --git a/TrainNemo/TrainGLaDOS_NeMo.ipynb b/TrainNemo/TrainGLaDOS_NeMo.ipynb index 1bfdeee..6325a7a 100644 --- a/TrainNemo/TrainGLaDOS_NeMo.ipynb +++ b/TrainNemo/TrainGLaDOS_NeMo.ipynb @@ -3,7 +3,7 @@ { "cell_type": "code", "execution_count": null, - "id": "b276bbe0", + "id": "90ce109a", "metadata": {}, "outputs": [], "source": [ @@ -21,7 +21,7 @@ { "cell_type": "code", "execution_count": null, - "id": "a9bb6563", + "id": "84df2778", "metadata": {}, "outputs": [], "source": [ @@ -52,7 +52,7 @@ { "cell_type": "code", "execution_count": null, - "id": "c3b7b7e3", + "id": "6c36b288", "metadata": {}, "outputs": [], "source": [ @@ -199,7 +199,7 @@ { "cell_type": "code", "execution_count": null, - "id": "82a7d268", + "id": "8c883c18", "metadata": {}, "outputs": [], "source": [ @@ -209,7 +209,7 @@ { "cell_type": "code", "execution_count": null, - "id": "832fcd67", + "id": "986e916e", "metadata": {}, "outputs": [], "source": [ @@ -220,7 +220,7 @@ { "cell_type": "code", "execution_count": null, - "id": "e56749bf", + "id": "ae039452", "metadata": {}, "outputs": [], "source": [ @@ -232,7 +232,7 @@ { "cell_type": "code", "execution_count": null, - "id": "9bf1fb18", + "id": "3717107a", "metadata": {}, "outputs": [], "source": [ @@ -256,7 +256,7 @@ { "cell_type": "code", "execution_count": null, - "id": "aefc57ff", + "id": "185a6f9b", "metadata": {}, "outputs": [], "source": [ @@ -273,7 +273,7 @@ { "cell_type": "code", "execution_count": null, - "id": "64b51e94", + "id": "1dd57280", "metadata": {}, "outputs": [], "source": [ @@ -288,7 +288,7 @@ { "cell_type": "code", "execution_count": null, - "id": "6b92a469", + "id": "98da0076", "metadata": {}, "outputs": [], "source": [ @@ -314,7 +314,7 @@ { "cell_type": "code", "execution_count": null, - "id": "780dba9a", + "id": "7ed81322", "metadata": {}, "outputs": [], "source": [ @@ -328,7 +328,7 @@ { "cell_type": "code", "execution_count": null, - "id": "831239e2", + "id": "56a8945f", "metadata": {}, "outputs": [], "source": [ @@ -384,7 +384,7 @@ { "cell_type": "code", "execution_count": null, - "id": "2fad0610", + "id": "7d6594e8", "metadata": {}, "outputs": [], "source": [ @@ -419,7 +419,7 @@ { "cell_type": "code", "execution_count": null, - "id": "0454967c", + "id": "e36967ad", "metadata": {}, "outputs": [], "source": [ @@ -509,7 +509,7 @@ { "cell_type": "code", "execution_count": null, - "id": "9d6dc1ee", + "id": "1178d4b7", "metadata": {}, "outputs": [], "source": [ @@ -531,7 +531,7 @@ { "cell_type": "code", "execution_count": null, - "id": "45085a4e", + "id": "741f10e2", "metadata": {}, "outputs": [], "source": [ @@ -577,7 +577,7 @@ { "cell_type": "code", "execution_count": null, - "id": "50b4c3be", + "id": "f75da634", "metadata": {}, "outputs": [], "source": [ @@ -590,7 +590,7 @@ { "cell_type": "code", "execution_count": null, - "id": "584dc353", + "id": "4dad4058", "metadata": {}, "outputs": [], "source": [ @@ -601,20 +601,49 @@ { "cell_type": "code", "execution_count": null, - "id": "52c6d0a9", + "id": "5d481c7b", "metadata": {}, "outputs": [], "source": [ - "# ngc registry resource download-version nvidia/riva/riva_quickstart:2.8.1\n", - "# pip3 install nema2riva whl\n", + "# pip3 install whl\n", + "# pip3 install nemo2riva\n", + "\n", + "# Run on training machine\n", "!nemo2riva --out hifigan.riva hifigan.nemo --key None\n", - "!nemo2riva --out fastpitch.riva fastpitch.nemo --key None" + "!nemo2riva --out fastpitch.riva fastpitch.nemo --key None\n", + "\n", + "# scp .riva files to jetson under /home/davesarmoury/RIVA/artifacts\n", + "# Everything below is run on the jetson\n", + "\n", + "# Servicemaker docker\n", + "docker run --gpus all -it --rm \\\n", + " -v /home/davesarmoury/RIVA/artifacts:/servicemaker-dev \\\n", + " -v /home/davesarmoury/RIVA/riva_repo:/data \\\n", + " --entrypoint=\\\"/bin/bash\\\" \\\n", + " nvcr.io/nvidia/riva/riva-speech:2.13.1-servicemaker-l4t-aarch64\n", + "\n", + "riva-build speech_synthesis \\\n", + " /servicemaker-dev/glados.rmir:tlt_encode \\\n", + " /servicemaker-dev/glados_fastpitch.riva:tlt_encode \\\n", + " /servicemaker-dev/glados_hifigan.riva:tlt_encode \\\n", + " --voice_name=GLaDOS \\\n", + " --sample_rate 22050\n", + "\n", + "riva-deploy /servicemaker-dev/glados.rmir:tlt_encode /data/models\n", + "\n", + "# Exit docker\n", + "\n", + "ngc registry resource download-version nvidia/riva/riva_quickstart\n", + "cd riva_quickstart\n", + "bash riva_init.sh\n", + "# copy glados.riva files into riva models dir\n", + "bash riva_start.sh" ] } ], "metadata": { "kernelspec": { - "display_name": "Python 3 (ipykernel)", + "display_name": "Python 3", "language": "python", "name": "python3" },