/UniBERT

Source code for "Unitary Multi-Margin BERT for Robust Natural Language Processing"

Primary LanguagePython

% Unitary Multi-Margin BERT (UniBERT) for Robust Natural Language Processing

About

This repository contains the source code for our paper "Unitary Multi-Margin BERT for Robust Natural Language Processing ."

Setup

To run simulations:

  1. install environments using conda from environment.yaml
  2. run attack/runsim.py

Note

This code requires a graphics card equivalent to NVidia 3090 with 24GB video RAM or above to run.