Skip to content
Permalink
Browse files

made requirements based on robustness lbirary

  • Loading branch information...
lengstrom committed Aug 21, 2019
1 parent 20e4d88 commit cbc2feb725d2075a8054f9cb0603004a0166f35b
Showing with 9 additions and 10 deletions.
  1. +1 −1 README.md
  2. +1 −0 requirements.txt
  3. +0 −1 robustness_lib
  4. +7 −8 user_constants.py
@@ -15,7 +15,7 @@ Steps to run the notebooks (for now, requires CUDA):
checkpoints there
- Install all the required packages with `pip install -r requirements.txt`
- Edit paths in `user_constants.py` to point to PyTorch-formatted versions of the `CIFAR` and `ImageNet` datasets
- Start a jupyter notebook server: `PYTHONPATH=robustness_lib/ jupyter notebook . --ip 0.0.0.0`
- Start a jupyter notebook server: `jupyter notebook . --ip 0.0.0.0`

## Citation

@@ -8,3 +8,4 @@ dill
tensorboardX
tables
tqdm
robustness
Submodule robustness_lib deleted from 48e8fd
@@ -1,10 +1,9 @@
# Path to dataset
DATA_PATH_DICT = {
'CIFAR': '/path/tools/cifar',
'RestrictedImageNet': '/path/tools/imagenet',
'ImageNet': '//path/tools/imagenet',
'H2Z': '/path/tools/horse2zebra',
'A2O': '/path/tools/apple2orange',
'S2W': '/path/tools/summer2winter_yosemite'
}

'CIFAR': '/tmp/',
'RestrictedImageNet': '/scratch/engstrom_scratch/imagenet',
'ImageNet': '/scratch/engstrom_scratch/imagenet',
'H2Z': '/scratch/datasets/A2B/horse2zebra',
'A2O': '/scratch/datasets/A2B/apple2orange',
'S2W': '/scratch/datasets/A2B/summer2winter_yosemite'
}

0 comments on commit cbc2feb

Please sign in to comment.
You can’t perform that action at this time.