From 5573c54217a445e5573d785e6c9d98c24598594a Mon Sep 17 00:00:00 2001
From: Christian Fuss <chrifuss@freenet.de>
Date: Tue, 25 Jun 2019 19:49:48 +0200
Subject: [PATCH] Made 'size' parameter for OneHot Layer optional.
---
src/main/resources/templates/gluon/elements/OneHot.ftl | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/main/resources/templates/gluon/elements/OneHot.ftl b/src/main/resources/templates/gluon/elements/OneHot.ftl
index 40991a3a..6e75def6 100644
--- a/src/main/resources/templates/gluon/elements/OneHot.ftl
+++ b/src/main/resources/templates/gluon/elements/OneHot.ftl
@@ -1,8 +1,8 @@
<#assign input = element.inputs[0]>
<#assign size = element.size>
<#if mode == "ARCHITECTURE_DEFINITION">
- self.${element.name} = OneHot(size=${size})
- <#include "OutputShape.ftl">
+ self.${element.name} = OneHot(size=${size})
+ <#include "OutputShape.ftl">
</#if>
<#if mode == "FORWARD_FUNCTION">
${element.name} = self.${element.name}(${input})
--
GitLab