conda env create -f env.yml
conda activate vect_compress
python train_vq.py --save-dir test-vq --vq-alpha 2.5 --codebook-size 512
python train_compai.py --save-dir test-compai --lambda 1e-2
check on models/autoencoder.py and edit the function: get_encoder_decoder