Skip to content
master
Switch branches/tags
Code

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
ptb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

div-norm

Code that implements paper "Normalizing the Normalizers: Comparing and Extending Network Normalization Schemes"

CIFAR experiments

  • First download the python version of CIFAR dataset here. https://www.cs.toronto.edu/~kriz/cifar.html

    • For CIFAR-10: Create a folder named "cifar-10" and move the uncompressed folder "cifar-10-batches-py" into "cifar-10".

    • For CIFAR-100: Create a folder named "cifar-100" and move the uncompressed folder "cifar-100-py" into "cifar-100".

  • Run training and testing:

python run_cifar_exp.py --dataset {cifar-10/cifar-100} --model {MODEL} --verbose

Replace MODEL with a pre-configured model name, e.g. "dn". For details, please take a look at cifar_exp_config.py.

PTB experiments

  • Run training and testing:
python run_ptb_exp.py --model {MODEL} --verbose

Replace MODEL with a pre-configured model name, see run_ptb_exp.py for details.

Super-resolution experiments

  • Download the datasets Set5, Set14, BSD200. Create a folder named "sr_data" and put uncompressed datasets as subfolders in it. Note that our code depends on h5py, cv2.

  • Generate the training and testing data by running gen_sr_data.m in Matlab (we used matlab's imresize function to generate training data which is named as, e.g., "data_X4.h5" in folder "sr_data"). You can easily modify the script to use your own training or testing data.

  • Run the following command to train and test the model. Please refer to the documentation in the beginning of file for more on the configurations.

python run_sr_exp.py --model dn --data_folder sr_data --verbose

Citation

If you use our code, please consider cite the following: Normalizing the Normalizer: Comparing and Extending Network Normalization Schemes. Mengye Ren, Renjie Liao, Raquel Urtasun, Fabian H. Sinz, Richard S. Zemel. ICLR 2017.

@inproceedings{ren17norm,
  author    = {Mengye Ren and Renjie Liao and Raquel Urtasun and Fabian H. Sinz and Richard S. Zemel},
  title     = {Normalizing the Normalizers: Comparing and Extending Network Normalization Schemes},
  booktitle = {ICLR},
  year      = {2017}
}

About

Implementation of divisive normalization in TensorFlow

Resources

Releases

No releases published

Packages

No packages published