Commit b0609627 authored by lr119628's avatar lr119628
Browse files

[update] formatting

parent 70d467b3
......@@ -297,6 +297,8 @@ class BLEU(mx.metric.EvalMetric):
return new_list
class ${tc.fileNameWithoutEnding}:
def __init__(self, data_loader, net_constructor):
self._data_loader = data_loader
......@@ -435,6 +437,7 @@ class ${tc.fileNameWithoutEnding}:
loss_function = LogCoshLoss()
else:
logging.error("Invalid loss parameter.")
loss_function.hybridize()
<#if tc.containsAdaNet()>
<#list tc.architecture.networkInstructions as networkInstruction>
......
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