README_PYTORCH.md

PyTorch Influence Functions

The PyTorch implementation is located in influence_pytorch/ and scripts_pytorch/.

1. Installation

conda create -n influence_pytorch python=3.8
conda activate influence_pytorch
pip install torch numpy scipy matplotlib seaborn pandas jupyter tqdm torchvision

2. Usage

Run the following commands in the scripts_pytorch directory:

cd scripts_pytorch

# 1. Run
python train_mnist_logreg.py

# 2. Generate Figure 2
python fig2_approx.py

The output figure will be saved to ../output_pytorch/fig2_approx_pytorch.png.