We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
batch_size
1 parent ff486c4 commit fe45f11Copy full SHA for fe45f11
tmva/sofie/test/generateKerasModels.py
@@ -28,7 +28,7 @@ def generateFunctionalModel():
28
29
def generateSequentialModel():
30
model=Sequential()
31
- model.add(Dense(8,batch_size=4))
+ model.add(Dense(8))
32
model.add(ReLU())
33
model.add(Dense(6))
34
model.add(Activation('sigmoid'))
@@ -43,7 +43,7 @@ def generateSequentialModel():
43
44
def generateBatchNormModel():
45
46
- model.add(Dense(4,batch_size=2))
+ model.add(Dense(4))
47
model.add(BatchNormalization())
48
model.add(Dense(2))
49
@@ -180,4 +180,4 @@ def generateCustomModel():
180
generateBinaryOpModel()
181
generateActivationModel()
182
generateSwishModel()
183
-generateCustomModel()
+generateCustomModel()
0 commit comments