diff --git a/README.md b/README.md index 114184e..53cf6cf 100644 --- a/README.md +++ b/README.md @@ -16,8 +16,14 @@ We propose *MultiDelete*, the first machine unlearning method that targets unlea

+## Installation +```bash +conda env create -n multidelete -f environment.yml +conda activate multidelete +``` + ## How to run 1. Step 1. Train original model diff --git a/environment.yml b/environment.yml new file mode 100644 index 0000000..d462dcb --- /dev/null +++ b/environment.yml @@ -0,0 +1,31 @@ +name: lavis-clip +channels: + - defaults + - conda-forge + - pytorch +dependencies: + - python=3.9 + - pip + - pip: + - torch==2.1.2+cu118 + - torchvision==0.16.2+cu118 + - torchaudio==2.1.2+cu118 + - --extra-index-url https://download.pytorch.org/whl/cu118 + - importlib_metadata + - omegaconf + - pandas + - numpy<2 + - iopath + - timm==0.6.13 + - decord + - webdataset + - opencv-python-headless + - transformers==4.30.2 + - fairscale==0.4.13 + - einops + - spacy + - pycocoevalcap + - sentencepiece + - wandb + - open_clip_torch +