Commit f3bc69ee authored by Svetlana Pavlitskaya's avatar Svetlana Pavlitskaya

added eval-metric

parent f78e68ea
Pipeline #66619 passed with stages
in 2 minutes and 37 seconds
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="VcsDirectoryMappings">
<mapping directory="" vcs="Git" />
</component>
</project>
\ No newline at end of file
......@@ -107,6 +107,7 @@ class ${tc.fileNameWithoutEnding}:
def train(self, batch_size,
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,
......
......@@ -107,6 +107,7 @@ class CNNCreator_Alexnet:
def train(self, batch_size,
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,
......
......@@ -107,6 +107,7 @@ class CNNCreator_CifarClassifierNetwork:
def train(self, batch_size,
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,
......
......@@ -107,6 +107,7 @@ class CNNCreator_VGG16:
def train(self, batch_size,
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,
......
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