Skip to content

Commit d3c18ef

Browse files
author
Hsiu-Wei Yang
committed
Fixed tutorial.
1 parent ad737fe commit d3c18ef

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

README.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff 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
```
2932
bash train_bert.sh 0 zh_en

train_bert.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,12 +10,12 @@ cd relogic
1010
python -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 \

0 commit comments

Comments
 (0)