• Haoyu Zhang's avatar
    Improve Keras graph performance for ResNet56 (#7241) · dd5a91d3
    Haoyu Zhang authored
    * Config threadpool, cuDNN persistent BN, and grappler layout optimizer properly for ResNet56
    
    * Add tweaked tests for Resnet56
    
    * Avoid triggering the last partial batch overhead by explicitly dropping remainder
    dd5a91d3
cifar10_main.py 10.3 KB