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

src/main/resources/templates/caffe2/CNNCreator.ftl

@@ -105,8 +105,9 @@ 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',
               optimizer_params=(('learning_rate', 0.001),),
               load_checkpoint=True,
@@ -158,6 +159,7 @@ class ${tc.fileNameWithoutEnding}:
 
         self.module.fit(
             train_data=train_iter,
+            eval_metric=eval_metric,
             eval_data=test_iter,
             optimizer=optimizer,
             optimizer_params=optimizer_params,

src/test/resources/target_code/CNNCreator_Alexnet.py

@@ -105,8 +105,9 @@ 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',
               optimizer_params=(('learning_rate', 0.001),),
               load_checkpoint=True,
@@ -158,6 +159,7 @@ class CNNCreator_Alexnet:
 
         self.module.fit(
             train_data=train_iter,
+            eval_metric=eval_metric,
             eval_data=test_iter,
             optimizer=optimizer,
             optimizer_params=optimizer_params,

src/test/resources/target_code/CNNCreator_CifarClassifierNetwork.py

@@ -105,8 +105,9 @@ 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',
               optimizer_params=(('learning_rate', 0.001),),
               load_checkpoint=True,
@@ -158,6 +159,7 @@ class CNNCreator_CifarClassifierNetwork:
 
         self.module.fit(
             train_data=train_iter,
+            eval_metric=eval_metric,
             eval_data=test_iter,
             optimizer=optimizer,
             optimizer_params=optimizer_params,

src/test/resources/target_code/CNNCreator_VGG16.py

@@ -105,8 +105,9 @@ 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',
               optimizer_params=(('learning_rate', 0.001),),
               load_checkpoint=True,
@@ -158,6 +159,7 @@ class CNNCreator_VGG16:
 
         self.module.fit(
             train_data=train_iter,
+            eval_metric=eval_metric,
             eval_data=test_iter,
             optimizer=optimizer,
             optimizer_params=optimizer_params,