add fig

OSU-NLP-Group · Oct 5, 2023 · b6ff2ee · b6ff2ee
1 parent 8dd924d
commit b6ff2ee
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -3,6 +3,8 @@ Code for the CIKM'23 paper "A Retrieve-and-Read Framework for Knowledge Graph Li
 
 <!-- Code will be released soon. -->
 
+![A Retrieve-and-Read Framework for Knowledge Graph Link Prediction (KG-R3)](./assets/KG-R3.png)
+
 # Install dependencies
 1. Create a new conda virtual env
 
@@ -60,7 +62,7 @@ python eval.py --dataset-path data/FB15K-237/ --cuda \
 
 ## WN18RR
 
-### train on 2 GPUs, per-GPU batch size=256
+### train, BS=512
 ```
 python -u main.py --dataset-path data/WN18RR/ --cuda \
 --save-dir ckpts/CKPT_DIR/ --sampling-type minerva \

diff --git a/assets/KG-R3.pdf b/assets/KG-R3.pdf