├── .gitignore ├── README.md ├── configs ├── config.yaml ├── data │ ├── amazon.yaml │ ├── cose.yaml │ ├── default.yaml │ ├── esnli.yaml │ ├── irony.yaml │ ├── movies.yaml │ ├── multirc.yaml │ ├── olid.yaml │ ├── sst.yaml │ ├── stf.yaml │ └── yelp.yaml ├── experiment │ ├── fixed_lm_expl.yaml │ └── iter_lm.yaml ├── hsearch │ └── lr.yaml ├── hydra │ └── default.yaml ├── logger │ └── neptune.yaml ├── model │ ├── a2r.yaml │ ├── expl_reg.yaml │ ├── fresh.yaml │ ├── lm.yaml │ ├── optimizer │ │ ├── adamw.yaml │ │ └── hf_adamw.yaml │ └── scheduler │ │ ├── fixed.yaml │ │ └── linear_with_warmup.yaml ├── setup │ └── a100.yaml ├── trainer │ └── defaults.yaml └── training │ ├── base.yaml │ ├── evaluate.yaml │ └── finetune.yaml ├── main.py ├── requirements.txt ├── scripts └── build_dataset.py └── src ├── __init__.py ├── data ├── __init__.py └── data.py ├── model ├── __init__.py ├── base_model.py ├── lm.py └── mlp.py ├── run.py └── utils ├── __init__.py ├── callbacks.py ├── conf.py ├── data.py ├── eraser ├── data_utils.py ├── metrics.py └── utils.py ├── expl.py ├── logging.py ├── losses.py ├── metrics.py ├── misc.py └── optim.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/README.md -------------------------------------------------------------------------------- /configs/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/config.yaml -------------------------------------------------------------------------------- /configs/data/amazon.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/data/amazon.yaml -------------------------------------------------------------------------------- /configs/data/cose.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/data/cose.yaml -------------------------------------------------------------------------------- /configs/data/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/data/default.yaml -------------------------------------------------------------------------------- /configs/data/esnli.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/data/esnli.yaml -------------------------------------------------------------------------------- /configs/data/irony.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/data/irony.yaml -------------------------------------------------------------------------------- /configs/data/movies.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/data/movies.yaml -------------------------------------------------------------------------------- /configs/data/multirc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/data/multirc.yaml -------------------------------------------------------------------------------- /configs/data/olid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/data/olid.yaml -------------------------------------------------------------------------------- /configs/data/sst.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/data/sst.yaml -------------------------------------------------------------------------------- /configs/data/stf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/data/stf.yaml -------------------------------------------------------------------------------- /configs/data/yelp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/data/yelp.yaml -------------------------------------------------------------------------------- /configs/experiment/fixed_lm_expl.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/experiment/fixed_lm_expl.yaml -------------------------------------------------------------------------------- /configs/experiment/iter_lm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/experiment/iter_lm.yaml -------------------------------------------------------------------------------- /configs/hsearch/lr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/hsearch/lr.yaml -------------------------------------------------------------------------------- /configs/hydra/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/hydra/default.yaml -------------------------------------------------------------------------------- /configs/logger/neptune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/logger/neptune.yaml -------------------------------------------------------------------------------- /configs/model/a2r.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/model/a2r.yaml -------------------------------------------------------------------------------- /configs/model/expl_reg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/model/expl_reg.yaml -------------------------------------------------------------------------------- /configs/model/fresh.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/model/fresh.yaml -------------------------------------------------------------------------------- /configs/model/lm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/model/lm.yaml -------------------------------------------------------------------------------- /configs/model/optimizer/adamw.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/model/optimizer/adamw.yaml -------------------------------------------------------------------------------- /configs/model/optimizer/hf_adamw.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/model/optimizer/hf_adamw.yaml -------------------------------------------------------------------------------- /configs/model/scheduler/fixed.yaml: -------------------------------------------------------------------------------- 1 | lr_scheduler: fixed 2 | warmup_updates: 0.0 3 | -------------------------------------------------------------------------------- /configs/model/scheduler/linear_with_warmup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/model/scheduler/linear_with_warmup.yaml -------------------------------------------------------------------------------- /configs/setup/a100.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/setup/a100.yaml -------------------------------------------------------------------------------- /configs/trainer/defaults.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/trainer/defaults.yaml -------------------------------------------------------------------------------- /configs/training/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/training/base.yaml -------------------------------------------------------------------------------- /configs/training/evaluate.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/training/evaluate.yaml -------------------------------------------------------------------------------- /configs/training/finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/configs/training/finetune.yaml -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/main.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/build_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/scripts/build_dataset.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/src/data/data.py -------------------------------------------------------------------------------- /src/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/model/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/src/model/base_model.py -------------------------------------------------------------------------------- /src/model/lm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/src/model/lm.py -------------------------------------------------------------------------------- /src/model/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/src/model/mlp.py -------------------------------------------------------------------------------- /src/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/src/run.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/src/utils/callbacks.py -------------------------------------------------------------------------------- /src/utils/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/src/utils/conf.py -------------------------------------------------------------------------------- /src/utils/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/src/utils/data.py -------------------------------------------------------------------------------- /src/utils/eraser/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/src/utils/eraser/data_utils.py -------------------------------------------------------------------------------- /src/utils/eraser/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/src/utils/eraser/metrics.py -------------------------------------------------------------------------------- /src/utils/eraser/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/src/utils/eraser/utils.py -------------------------------------------------------------------------------- /src/utils/expl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/src/utils/expl.py -------------------------------------------------------------------------------- /src/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/src/utils/logging.py -------------------------------------------------------------------------------- /src/utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/src/utils/losses.py -------------------------------------------------------------------------------- /src/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/src/utils/metrics.py -------------------------------------------------------------------------------- /src/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/src/utils/misc.py -------------------------------------------------------------------------------- /src/utils/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/UNIREX/HEAD/src/utils/optim.py --------------------------------------------------------------------------------