run_book.sh

CUDA_VISIBLE_DEVICES=2 python main_pretrain.py --gat_split 10 --model_name UGAT_mlp --data_folder /gaolehe/data/kb_final/ --dataset book --batch_size 4096 --embedding_size 100 --n_epochs 1000 --lr 1e-3 --decay_rate 0.99 --checkpoint_dir checkpoint/good_pretrain/book --n_sample 200 --l2_lambda 1e-5 --rs_sample 20 --rs_sample_flag --kg_sample 10 --kg_sample_flag --eval_every 5 --experiment_name book-ugat-mlp-200sample-norm-emb --query_weight --load_ckpt_file init/book-DistMult-bce-decay.ckpt --norm_emb
CUDA_VISIBLE_DEVICES=1 python main_upgan.py --gat_split 10 --model_name UGAT_mlp --G_name generator_concat --data_folder /gaolehe/data/kb_final/ --dataset book --batch_size 4096 --embedding_size 100 --n_epochs 50 --lr 1e-4 --lr_g 1e-4 --decay_rate 0.0 --checkpoint_dir checkpoint/upgan --n_sample 1024 --n_sample_gen 200 --l2_lambda 1e-5 --l2_lambda_g 1e-5 --rs_sample 20 --rs_sample_flag --kg_sample 10 --kg_sample_flag --eval_every 3 --experiment_name book-norm_ent-concat-noise --query_weight --load_ckpt_file good_pretrain/book/book-ugat-mlp-200sample-norm-emb.ckpt --load_ckpt_G init/book-DistMult-bce-decay.ckpt --norm_emb --reward_type baseline-softmax --lambda_smooth 0.01 --sigma 1.0