Predicting Inductive Biases of Pre-Trained Models
@inproceedings{
lovering2021predictinginductive,
title={Predicting Inductive Biases of Fine-tuned Models},
author={Charles Lovering and Rohan Jha and Tal Linzen and Ellie Pavlick},
booktitle={International Conference on Learning Representations},
year={2021},
url={https://openreview.net/forum?id=mNtmhaDkAr}
}
Setup
Install requirements.
# Create new env.
conda create --name features python=3.8
conda activate features
# Install pytorch.
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
# Install further reqs.
pip install pytest tqdm pandas gputil spacy[cuda102] transformers pytorch_lightning pyinflect sklearn wandb nltk
pip install plac --upgrade
python -m spacy download en_core_web_lg
Set wandb
subscription key in your ~/.bash_profile
.
# This is not the real key.
export WANDB_API_KEY=628318530717958647692528
Generate experiments & run!
# generate datasets
./setup.sh
# approx <30 min
sbatch datasets.sh
pytest test.py
# generate jobs
python job.py --experiment finetune
python job.py --experiment probing
# run jobs
sbatch jobs/[DATE]/jobs.sh
Troubleshooting
If you have issues with plac
(e.g. plac.opt
is not defined) reinstall it with pip install plac --upgrade
.
If you have issues with cupy
uninstall (pip uninstall cupy-cuda102
) and then re-install (pip install cupy-cuda102
).
Let us (@cjlovering, @rohjha) know if you have any questions.