I m not sure if this is a problem with pytorch or just a problem with my GPU or device or code. However the same code did run without error on another 1080ti device with same Linux, python, pytorch, cuda and cudnn version.
For me, i just did:
git clone git@github.com:zwlanpishu/tacotron2.git
git checkout baseline_test
python3 train.py --output_directory “yourpath” --log_directory “yourpath”
The data can be download from https://keithito.com/LJ-Speech-Dataset/
The txt files in the “filelists” dirrectory should be rewrite from “DUMMY/xxx” to “your data path/xxx”.
my env:
Ubuntu 18.04
python 3.6.9
pytorch 1.5.0
cuda 10.2
cudnn 7.6.5.32
The other pip dependencys are listed in requirements.txt, while tensorflow is not needed.