• Karmel Allison's avatar
    Add SavedModel export to Resnet (#3759) · eb73a850
    Karmel Allison authored
    * Adding export_dir and model saving for Resnet
    
    * Moving to utils for tests
    
    * Adding batch_size
    
    * Adding multi-gpu export warning
    
    * Responding to CR
    
    * Py3 compliance
    eb73a850
export.py 1.88 KB