python -m tools.run_linear_probe \
--seed 0 \
--dataset cifar \
--out-dir logs/checkpoints/cifar10-r18-seed0/linear_probe/remove_50_random_neurons_seed0 \
--adv-train 0 \
--adv-eval 0 \
--arch resnet18 \
--batch-size 512 \
--lr 0.0001 \
--epochs 40 \
--num_classes 10 \
--nb_neurons 50 \
--resume logs/checkpoints/cifar10-r18-seed0/checkpoint.pt.best
