Imported from https://github.com/dalgu90/wrn-tensorflow