├── .gitignore
├── LICENSE
├── README.md
├── hqa.ipynb
├── requirements.txt
└── setup.sh


/.gitignore:
--------------------------------------------------------------------------------
1 | venv_hqa
2 | models
3 | __pycache__
4 | **/model.pt*
5 | .ipynb_checkpoints
6 | 


--------------------------------------------------------------------------------
/LICENSE:
--------------------------------------------------------------------------------
 1 | MIT License
 2 | 
 3 | Copyright (c) 2020 Speechmatics
 4 | 
 5 | Permission is hereby granted, free of charge, to any person obtaining a copy
 6 | of this software and associated documentation files (the "Software"), to deal
 7 | in the Software without restriction, including without limitation the rights
 8 | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 9 | copies of the Software, and to permit persons to whom the Software is
10 | furnished to do so, subject to the following conditions:
11 | 
12 | The above copyright notice and this permission notice shall be included in all
13 | copies or substantial portions of the Software.
14 | 
15 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21 | SOFTWARE.
22 | 


--------------------------------------------------------------------------------
/README.md:
--------------------------------------------------------------------------------
 1 | # hqa
 2 | Code to accompany the paper "Hierarchical Quantized Autoencoders"
 3 | 
 4 | ## Installation
 5 | 
 6 | ```
 7 | $ ./setup.sh
 8 | $ source venv_hqa/bin/activate
 9 | $ jupyter lab # open hqa.ipynb
10 | ```


--------------------------------------------------------------------------------
/requirements.txt:
--------------------------------------------------------------------------------
1 | torch==1.3.1
2 | torchvision==0.4.2
3 | matplotlib==3.1.1
4 | numpy==1.17.2
5 | jupyter==1.0.0
6 | jupyterlab==1.2.21
7 | absl-py==0.8.1
8 | Pillow==9.0.1


--------------------------------------------------------------------------------
/setup.sh:
--------------------------------------------------------------------------------
 1 | #!/bin/bash -e
 2 | 
 3 | VENV=./venv_hqa
 4 | 
 5 | if [ ! -d $VENV ]; then
 6 |     echo "Creating $VENV"
 7 |     virtualenv -p python3.7 $VENV
 8 | fi
 9 | 
10 | echo "Starting venv..."
11 | source $VENV/bin/activate
12 | 
13 | echo "Installing requirements..."
14 | 
15 | pip3 install -r ./requirements.txt
16 | 


--------------------------------------------------------------------------------