Commit d0a77cf8 authored by Carlos Alfredo Yeverino Rodriguez's avatar Carlos Alfredo Yeverino Rodriguez
Browse files

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

parent cf735f54
......@@ -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