#!/usr/bin/env bash

set -x

echo "Running hyperbolic VAE with TangentRealNVP flow for Cora: changing dimension"

python main.py --dataset='cora' --model='hyperbolic' --flow_model='TangentRealNVP' --epochs=3000 --hidden_dim=128 --flow_hidden_size=64 --z_dim=2 --wandb --namestr='hyperbolic-d2-TangentRealNVP'
python main.py --dataset='cora' --model='hyperbolic' --flow_model='TangentRealNVP' --epochs=3000 --hidden_dim=128 --flow_hidden_size=64 --z_dim=4 --wandb --namestr='hyperbolic-d4-TangentRealNVP'
python main.py --dataset='cora' --model='hyperbolic' --flow_model='TangentRealNVP' --epochs=3000 --hidden_dim=128 --flow_hidden_size=64 --z_dim=6 --wandb --namestr='hyperbolic-d6-TangentRealNVP'
python main.py --dataset='cora' --model='hyperbolic' --flow_model='TangentRealNVP' --epochs=3000 --hidden_dim=128 --flow_hidden_size=64 --z_dim=8 --wandb --namestr='hyperbolic-d8-TangentRealNVP'
python main.py --dataset='cora' --model='hyperbolic' --flow_model='TangentRealNVP' --epochs=3000 --hidden_dim=128 --flow_hidden_size=64 --z_dim=10 --wandb --namestr='hyperbolic-d10-TangentRealNVP'

echo "Running hyperbolic VAE with TangentRealNVP flow for Cora: changing number of layers"

python main.py --dataset='cora' --model='hyperbolic' --flow_model='TangentRealNVP' --epochs=3000 --hidden_dim=128 --flow_hidden_size=64 --n_blocks=2 --wandb --namestr='hyperbolic-l2-TangentRealNVP'
python main.py --dataset='cora' --model='hyperbolic' --flow_model='TangentRealNVP' --epochs=3000 --hidden_dim=128 --flow_hidden_size=64 --n_blocks=4 --wandb --namestr='hyperbolic-l4-TangentRealNVP'
python main.py --dataset='cora' --model='hyperbolic' --flow_model='TangentRealNVP' --epochs=3000 --hidden_dim=128 --flow_hidden_size=64 --n_blocks=6 --wandb --namestr='hyperbolic-l6-TangentRealNVP'
python main.py --dataset='cora' --model='hyperbolic' --flow_model='TangentRealNVP' --epochs=3000 --hidden_dim=128 --flow_hidden_size=64 --n_blocks=8 --wandb --namestr='hyperbolic-l8-TangentRealNVP'
python main.py --dataset='cora' --model='hyperbolic' --flow_model='TangentRealNVP' --epochs=3000 --hidden_dim=128 --flow_hidden_size=64 --n_blocks=10 --wandb --namestr='hyperbolic-l10-TangentRealNVP'
