Set default value for batch_size in CNNCreator.ftl since training parameters can be empty.

parent cf735f54
Pipeline #69467 passed with stages
in 1 minute and 39 seconds
......@@ -105,7 +105,7 @@ class ${tc.fileNameWithoutEnding}:
sys.exit(1)
def train(self, batch_size,
def train(self, batch_size=64,
num_epoch=10,
eval_metric='acc',
optimizer='adam',
......
......@@ -105,7 +105,7 @@ class CNNCreator_Alexnet:
sys.exit(1)
def train(self, batch_size,
def train(self, batch_size=64,
num_epoch=10,
eval_metric='acc',
optimizer='adam',
......
......@@ -105,7 +105,7 @@ class CNNCreator_CifarClassifierNetwork:
sys.exit(1)
def train(self, batch_size,
def train(self, batch_size=64,
num_epoch=10,
eval_metric='acc',
optimizer='adam',
......
......@@ -105,7 +105,7 @@ class CNNCreator_VGG16:
sys.exit(1)
def train(self, batch_size,
def train(self, batch_size=64,
num_epoch=10,
eval_metric='acc',
optimizer='adam',
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment