THU-BPM / PROMU

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

  • Run the full model on NewsCLIPpings dataset with default hyperparameter settings

sh run.sh

Data Download

  • NewsCLIPpings: Automatic Generation of Out-of-Context Multimodal Media (access)

About

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


Languages

Language:Python 99.9%Language:Shell 0.1%