Skip to content

Commit

Permalink
Merge pull request #7 from iamgroot42/michael/tprfpr_fix
Browse files Browse the repository at this point in the history
Updates to temporal MIA scripts
  • Loading branch information
michaelduan8 committed Feb 14, 2024
2 parents 26f77aa + a84defa commit 4789787
Show file tree
Hide file tree
Showing 3 changed files with 18 additions and 21 deletions.
7 changes: 3 additions & 4 deletions scripts/run_mia_suite_temporal_arxiv.sh
Original file line number Diff line number Diff line change
@@ -1,13 +1,12 @@
#!/bin/bash
version=unified_mia_v5_temporal_arxiv_v2_ne
version=unified_mia_v5_temporal_arxiv_v3_ne

for source in "arxiv_2020-08" "arxiv_2021-01" "arxiv_2021-06" "arxiv_2022-06" "arxiv_2023-06" "arxiv_2023-01" "arxiv_2022-01" #"arxiv_2019-01" "arxiv_2020-01"
do
python run.py \
--experiment_name $version \
--config configs/single_gpu_mi.json \
--base_model "EleutherAI/pythia-12b-deduped" \
--revision step99000 \
--specific_source $source \
--output_name $version \
--baselines_only true
--specific_source $source
done
15 changes: 15 additions & 0 deletions scripts/run_mia_suite_temporal_wiki.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
#!/bin/bash
version=unified_mia_v5_temporal_wiki_v2_ne

for subset in "temporal_wiki_full"
do
for model in "pythia-1.4b-deduped" "pythia-160m-deduped" "pythia-2.8b-deduped" "pythia-6.9b-deduped" #"pythia-12b-deduped"
do
python run.py \
--experiment_name $version \
--config configs/mi.json \
--base_model "EleutherAI/${model}" \
--revision step99000 \
--specific_source ${subset}
done
done
17 changes: 0 additions & 17 deletions scripts/run_mia_suite_wikiMIA.sh

This file was deleted.

0 comments on commit 4789787

Please sign in to comment.