#!/bin/bash
data_dir=''
graph_path=''
data_prefix='vax'
out_dir=''


max_seq_length=128
encoder_type='ATTN'
history_size=64
mark_embedding_size=64

n_components=32
seed=1
batch_size=256

max_epochs=1000
max_loop=1
patience=10
save_freq=10
gpu0sz=46

python ../amdn_Gaussian-tied-unco-new-EM-NN-KL-vax-multiGPU.py --data_dir=$data_dir --data_prefix=$data_prefix \
--out_dir=$out_dir --graph_path $graph_path --max_seq_length=$max_seq_length \
--encoder_type=$encoder_type --history_size=$history_size --mark_embedding_size=$mark_embedding_size  \
--n_components=$n_components --seed=$seed --batch_size $batch_size --max_epochs $max_epochs \
--patience $patience --save_freq $save_freq --gpu0sz $gpu0sz --max_loop $max_loop

