official/resnet/keras/keras_imagenet_main.py · 1cdc35c8a57e18e1e0dc7137db33cc667c922bce · ModelZoo / ResNet50_tensorflow

Handle data format and graph compile properly for better GPU performance (#6013) · 1cdc35c8

Haoyu Zhang authored Jan 09, 2019

* Handle data format in Keras ResNet model properly for better performance on GPU; Compile only the training graph when skip_eval flag is True

* Added data format fix to Keras Cifar model; Removed unnecessary import

* Add a comment to the skip_eval flag per Priya's request

1cdc35c8

keras_imagenet_main.py 6.54 KB

Replace keras_imagenet_main.py