artikandri / multitask_nlp

Multi-task model for classification tasks, built with PyTorch

Multitask NLP

Multitask model built with PyTorch

Configure environment

Install requirements

pip install -r requirements.txt

Download data

Pull data from DVC:

dvc pull

Some data are in archives, run script which unzips them

python multitask_nlp\scripts\unzip_data.py

Experiments

Run experiments

python -m multitask_nlp.experiments.multitask

Single task experiements are located in multitask_nlp/experiments/single_task_exp so for example run them as

python -m multitask_nlp.experiments.single_task_exp.boolq