File tree Expand file tree Collapse file tree 2 files changed +5
-2
lines changed
Expand file tree Collapse file tree 2 files changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -24,6 +24,9 @@ bash graph.sh 0 0
2424```
2525
2626## Run PairwiseBERT
27+ We use the package ` relogic ` to derive BERT-based embeddings.
28+ Please install ` relogic ` before you run the followings.
29+ Note that the argument ` --local_rank ` in ` relogic ` indicates your gpu id.
2730#### Training
2831```
2932bash train_bert.sh 0 zh_en
Original file line number Diff line number Diff line change @@ -10,12 +10,12 @@ cd relogic
1010python -u -m relogic.main \
1111 --task_name pairwise \
1212 --mode train \
13- --output_dir saves/pair_matching/$2 \ # project name
13+ --output_dir saves/pair_matching/$2 \
1414 --bert_model bert-base-multilingual-cased \
1515 --raw_data_path data/raw_data/entity_alignment \
1616 --label_mapping_path none \
1717 --model_name default \
18- --local_rank $1 \ # GPU id
18+ --local_rank $1 \
1919 --train_batch_size 3 \
2020 --test_batch_size 3 \
2121 --learning_rate 1e-6 \
You can’t perform that action at this time.
0 commit comments