# CUDA_VISIBLE_DEVICES=6 python -u code/run.py \
#  --do_train \
#  --cuda \
#  --do_valid \
#  --do_test \
#  --data_path data \
#  --process_path processed_data \
#  --dataset PrimeKG \
#  --model ComplEx \
#  --distance_metric cosine \
#  --hierarchy_type llm \
#  --zeta_3 8.0 \
#  -n 512 -b 512 -d 1024 \
#  -g 200.0 -a 1.0 -adv \
#  -lr 0.00001 --max_steps 300000 -r 0.000005 \
#  --test_batch_size 8 -de -dr \
#  --text_dist_constraint true \
#  --hier_dist_constraint false \
#  --intra_cluster_constraint true \
#  --inter_cluster_constraint true \


#  CUDA_VISIBLE_DEVICES=5 python -u code/run.py \
#  --do_train \
#  --cuda \
#  --do_valid \
#  --do_test \
#  --data_path data \
#  --dataset YAGO3-10 \
#  --model ComplEx \
#  --distance_metric cosine \
#  --hierarchy_type llm \
#  --zeta_3 9.0 \
#  -n 256 -b 256 -d 512 \
#  -g 24.0 -a 1.0 -adv \
#  -lr 0.00001 --max_steps 800000 \
#  --test_batch_size 8 -de -dr\
#  --text_dist_constraint true \
#  --hier_dist_constraint false \
#  --intra_cluster_constraint true \
#  --inter_cluster_constraint true \

#  CUDA_VISIBLE_DEVICES=5 python -u code/run.py \
#  --do_train \
#  --cuda \
#  --do_valid \
#  --do_test \
#  --data_path data \
#  --dataset FB15K-237 \
#  --model ComplEx \
#  --distance_metric cosine \
#  --hierarchy_type llm \
#  --zeta_3 3.5 \
#  -n 256 -b 256 -d 1024 \
#  -g 200.0 -a 1.0 -adv \
#  -lr 0.0001 --max_steps 400000 \
#  --test_batch_size 8 \
#  --text_dist_constraint true \
#  --hier_dist_constraint false \
#  --intra_cluster_constraint true \
#  --inter_cluster_constraint true \
#  --valid_steps 20000  -de -dr \