Fix for missing quotes in eval_metric value

parent d0a77cf8
Pipeline #69473 passed with stages
in 1 minute and 41 seconds
......@@ -29,7 +29,7 @@ if __name__ == "__main__":
normalize = ${config.normalize?string("True","False")},
</#if>
<#if (config.evalMetric)??>
eval_metric = ${config.evalMetric},
eval_metric = '${config.evalMetric}',
</#if>
<#if (config.configuration.optimizer)??>
optimizer = '${config.optimizerName}',
......
......@@ -16,7 +16,7 @@ if __name__ == "__main__":
load_checkpoint = True,
context = 'gpu',
normalize = True,
eval_metric = mse,
eval_metric = 'mse',
optimizer = 'rmsprop',
optimizer_params = {
'weight_decay': 0.01,
......@@ -40,7 +40,7 @@ if __name__ == "__main__":
load_checkpoint = False,
context = 'gpu',
normalize = False,
eval_metric = topKAccuracy,
eval_metric = 'topKAccuracy',
optimizer = 'adam',
optimizer_params = {
'epsilon': 1.0E-6,
......
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