the code for "FastERE: A Fast Framework for Entity Relation Extraction"
WIP warning: This is a preview version of our work.
For Training
python main.py
numpy==1.24.2
pytorch_crf==0.7.2
pytorch_lightning==1.9.4
PyYAML==6.0
torch==2.0.0
torchcrf==1.1.0
transformers==4.27.1
wandb
gpustat
tensorboardX