Skip to content

Commit 995caab

Browse files
Satish615Satish Gollaprolu
andauthored
Fix sm integration issues (#359)
* use tensorflow.keras instead of standalone keras library * use tf keras for inception scripts Co-authored-by: Satish Gollaprolu <sgollapr@amazon.com>
1 parent 8800bcf commit 995caab

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

test/resources/keras_inception.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
import argparse
1414
import os
1515

16-
import keras
16+
from tensorflow import keras
1717
import tensorflow as tf
1818

1919
parser = argparse.ArgumentParser()

test/resources/mnist/horovod_mnist.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -15,14 +15,14 @@
1515
import os
1616
import subprocess
1717

18-
import keras
19-
from keras.datasets import mnist
20-
from keras.models import Sequential
21-
from keras.layers import Dense, Dropout, Flatten
22-
from keras.layers import Conv2D, MaxPooling2D
23-
from keras import backend as K
18+
from tensorflow import keras
19+
from tensorflow.keras.datasets import mnist
20+
from tensorflow.keras.models import Sequential
21+
from tensorflow.keras.layers import Dense, Dropout, Flatten
22+
from tensorflow.keras.layers import Conv2D, MaxPooling2D
23+
from tensorflow.keras import backend as K
2424
import tensorflow as tf
25-
import horovod.keras as hvd
25+
import horovod.tensorflow.keras as hvd
2626

2727

2828
# Horovod: initialize Horovod.

0 commit comments

Comments
 (0)