To install requirements:
conda env create -f environment. yml
To train the model in the paper, run this command:
python src/main.py --mode train --batch 10 --accumulate_grad_batches 50
To generate results on test data, run:
python src/main.py --mode predict --checkpoint_path path_to_chkpt