• Toby Boyd's avatar
    Imagenet short tests (#6136) · 23721cf2
    Toby Boyd authored
    * Add short imagenet tests (taken from seemuch)
    - also rename to match go forward naming
    
    * fix method name
    
    * Update doc strings.
    
    * Fixe gpu number.
    
    * Add fill_objects
    
    * fixed calling wrong class in super.
    
    * fix lint issue.
    23721cf2
keras_imagenet_benchmark.py 5.59 KB