THU-BPM / PROMU

Source code of paper:"Prompt Me Up: Unleashing the Power of Alignments for Multimodal Entity and Relation Extraction".
15 stars 0 forks source link

PROMU

The source code of paper "Prompt Me Up: Unleashing the Power of Alignments for Multimodal Entity and Relation Extraction"

Installation

For training, a GPU is recommended to accelerate the training speed.

PyTroch

The code is based on PyTorch 1.6+. You can find tutorials here.

Dependencies

The dependencies are summarized in the file requirements.txt.

You can install these dependencies like this:

pip3 install -r requirements.txt

You also need to install Horovod and Apex

Usage

sh run.sh

Data Download