ref: 1f53f1e0a9b1e055222b28a70b2e327787e50d09
dir: /dnn/torch/rdovae/README.md/
# Rate-Distortion-Optimized Variational Auto-Encoder ## Setup The python code requires python >= 3.6 and has been tested with python 3.6 and python 3.10. To install requirements run ``` python -m pip install -r requirements.txt ``` ## Training To generate training data use dump date from the main LPCNet repo ``` ./dump_data -train 16khz_speech_input.s16 features.f32 data.s16 ``` To train the model, simply run ``` python train_rdovae.py features.f32 output_folder ``` To train on CUDA device add `--cuda-visible-devices idx`. ## ToDo - Upload checkpoints and add URLs