X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;ds=sidebyside;f=vnfs%2FDAaaS%2Fsample-apps%2Ftraining%2Fsample-horovod-app%2Fkeras_mnist_advanced_modified.py;fp=vnfs%2FDAaaS%2Fsample-apps%2Ftraining%2Fsample-horovod-app%2Fkeras_mnist_advanced_modified.py;h=7cfa7045f9374e1dd9fdd2cbe6799c7168fe9ffe;hb=b8c5101a920275687fdab980048cb4dae5aa902d;hp=fa39cb6a2c9e31633dd01b2fd3d5061793b34dc0;hpb=00107ed0ee9911ac761a544bada794a1ebb26ddf;p=demo.git diff --git a/vnfs/DAaaS/sample-apps/training/sample-horovod-app/keras_mnist_advanced_modified.py b/vnfs/DAaaS/sample-apps/training/sample-horovod-app/keras_mnist_advanced_modified.py index fa39cb6a..7cfa7045 100644 --- a/vnfs/DAaaS/sample-apps/training/sample-horovod-app/keras_mnist_advanced_modified.py +++ b/vnfs/DAaaS/sample-apps/training/sample-horovod-app/keras_mnist_advanced_modified.py @@ -1,5 +1,4 @@ from __future__ import print_function -import keras import os from tensorflow.keras.datasets import mnist from tensorflow.keras.models import Sequential @@ -7,10 +6,6 @@ from tensorflow.keras.layers import Dense, Dropout, Flatten from tensorflow.keras.layers import Conv2D, MaxPooling2D from tensorflow.keras.preprocessing.image import ImageDataGenerator from tensorflow.keras import backend as K -from tensorflow_estimator.python.estimator.export import export as export_helpers -from tensorflow.python.saved_model import builder as saved_model_builder -from tensorflow.python.saved_model import tag_constants, signature_constants -from tensorflow.python.saved_model.signature_def_utils_impl import predict_signature_def import tensorflow as tf import horovod.tensorflow.keras as hvd