• Haoyu Zhang's avatar
    Enable CuDNN BatchNorm spatial persistent by default (#6710) · 58deb059
    Haoyu Zhang authored
    * Enable CuDNN BatchNorm spatial persistent by default; Remove 2nd zero padding layer
    
    * Apply scale=False and fused=True consistently to BatchNorm layers
    
    * Undo remove padding layer
    
    * Replace zero padding with padding attribute in max pooling for better performance
    
    * Resolve comments
    
    * Revert "Replace zero padding with padding attribute in max pooling for better performance"
    
    This reverts commit ad49db057c800ecac008eec1057005bd2c08ac73.
    58deb059
resnet_model.py 10 KB