• Toby Boyd's avatar
    Add resnet56 short tests. (#6101) · 2519f29b
    Toby Boyd authored
    * Add resnet56 short tests.
    - created base benchmark module
    - renamed accuracy test class to contain the word Accuracy
    which will result in a need to update all the jobs
    and a loss of history but is worth it.
    - short tests are mostly copied from shining with oss refactor
    
    * Address feedback.
    
    * Move flag_methods to init
    - Address setting default flags repeatedly.
    
    * Rename accuracy tests.
    
    * Lint errors resolved.
    
    * fix model_dir set to flags.data_dir.
    
    * fixed not fulling pulling out flag_methods.
    2519f29b
keras_cifar_benchmark.py 6.99 KB