We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ba0c012 commit 31adb8dCopy full SHA for 31adb8d
1 file changed
tmva/sofie/test/generateKerasModels.py
@@ -105,6 +105,7 @@ def generateConv2DModel_SamePadding_Dilated():
105
model.fit(x_train, y_train, verbose=0, epochs=10, batch_size=2)
106
model.save("KerasModelConv2D_SameDilated.keras")
107
108
+
109
def generateReshapeModel():
110
model = Sequential()
111
model.add(Conv2D(8, kernel_size=3, activation="relu", input_shape=(4,4,1), padding="same"))
0 commit comments