Imported from https://github.com/convnets-as-gps/convnets-as-gps