• Taylor Robie's avatar
    Resnet distribution strategies (#3887) · 32aa6563
    Taylor Robie authored
    * begin transfer from contrib fork
    
    more changes to resnet_run_loop
    
    use AUTOTUNE in prefetch
    
    first pass at resnet with functional distribution strategies
    
    fix syntax error
    
    delint
    
    aesthetic tweaks
    
    delint and fix typos
    
    rip multi_gpu flag out of resnet entirely. Subject to saved model load verification
    
    update cifar10 and imagenet tests to reflect that the model function no longer need to know about multi_gpu
    
    fix imagenet test
    
    start addressing PR comments
    
    more PR response work
    
    * misc tweaks
    
    * add a comment
    
    * final pr tweaks
    
    * fix parsers
    32aa6563
resnet_run_loop.py 19.1 KB