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

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__": ...@@ -29,7 +29,7 @@ if __name__ == "__main__":
normalize = ${config.normalize?string("True","False")}, normalize = ${config.normalize?string("True","False")},
</#if> </#if>
<#if (config.evalMetric)??> <#if (config.evalMetric)??>
eval_metric = ${config.evalMetric}, eval_metric = '${config.evalMetric}',
</#if> </#if>
<#if (config.configuration.optimizer)??> <#if (config.configuration.optimizer)??>
optimizer = '${config.optimizerName}', optimizer = '${config.optimizerName}',
......
...@@ -16,7 +16,7 @@ if __name__ == "__main__": ...@@ -16,7 +16,7 @@ if __name__ == "__main__":
load_checkpoint = True, load_checkpoint = True,
context = 'gpu', context = 'gpu',
normalize = True, normalize = True,
eval_metric = mse, eval_metric = 'mse',
optimizer = 'rmsprop', optimizer = 'rmsprop',
optimizer_params = { optimizer_params = {
'weight_decay': 0.01, 'weight_decay': 0.01,
...@@ -40,7 +40,7 @@ if __name__ == "__main__": ...@@ -40,7 +40,7 @@ if __name__ == "__main__":
load_checkpoint = False, load_checkpoint = False,
context = 'gpu', context = 'gpu',
normalize = False, normalize = False,
eval_metric = topKAccuracy, eval_metric = 'topKAccuracy',
optimizer = 'adam', optimizer = 'adam',
optimizer_params = { optimizer_params = {
'epsilon': 1.0E-6, '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