train_wn18.sh

#!/usr/bin/env bash
CUDA_VISIBLE_DEVICES=0 \
python TrainMain.py \
--data_dir ../data/WN18/ \
--embedding_dim 50 \
--margin_value 4 \
--batch_size 3000 \
--learning_rate 0.01 \
--n_generator 24 \
--n_rank_calculator 24 \
--eval_freq 100 \
--max_epoch 5000