WIP
$ poetry install
$ poetry run pip install torch===1.5.1 torchvision===0.6.1 -f https://download.pytorch.org/whl/torch_stable.html
$ poetry run pip install pytorch_lightning test-tube$ poetry run python train.py -c configs/<config-file-name.yaml>e.g. poetry run python train.py -c configs/vae.yaml
$ poetry run python visualize.py configs/<config-file-name.yaml> <experiment_idx>e.g. poetry run python visualize.py configs/vae.yaml 1
ImageVisualizer shows latent variables sorted with each importance.
In this example, 45 means the index of latent variables, (105.455) means the importance and 0.0 means the variable.
You can adjust latent variables as follow:
