Skip to content

Latest commit

 

History

History
executable file
·
31 lines (24 loc) · 1023 Bytes

README.md

File metadata and controls

executable file
·
31 lines (24 loc) · 1023 Bytes

LIMIT-BERT

Contents

  1. Requirements
  2. Training

Requirements

  • Python 3.6 or higher.
  • Cython 0.25.2 or any compatible version.
  • PyTorch 1.0.0+.
  • EVALB. Before starting, run make inside the EVALB/ directory to compile an evalb executable. This will be called from Python for evaluation.
  • pytorch-transformers PyTorch 1.0.0+ or any compatible version.

Pre-trained Models (PyTorch)

The following pre-trained models are available for download from Google Drive:

Training

To train LIMIT-BERT, simply run:

sh run_limitbert.sh

Evaluation Instructions

To test after setting model path:

sh test_bert.sh