diff --git a/src/main/resources/templates/caffe2/elements/Input.ftl b/src/main/resources/templates/caffe2/elements/Input.ftl
index c2f0136939fd4d50bee33ff22f3de4d0dc739d18..1883500ba4a3fcac6d72993ed9ca3da43e18b4e8 100644
--- a/src/main/resources/templates/caffe2/elements/Input.ftl
+++ b/src/main/resources/templates/caffe2/elements/Input.ftl
@@ -9,11 +9,10 @@
     		${element.name} = data
 <#include "OutputShape.ftl">
 <#if heightIndex != channelIndex + 1 || widthIndex != heightIndex + 1>
-    		${element.name} = mx.symbol.transpose(data=${element.name},mx.sym.var    <#-- TODO: check how to adapt CNNArchLang transpose case -->
-    		    axes=(0,${tc.join(indexList, ",")}))
+    		${element.name} = model.net.Transpose(${element.name}, '${element.name}', axes=[0,${tc.join(indexList, ",")}])
 
 </#if>
 <#if indexList?size != 3>
-    		${element.name} = mx.symbol.reshape(data=${element.name},    <#-- TODO: check how to adapt CNNArchLang transpose case -->
+    		${element.name}, _ = model.net.Reshape('${element.name}', ['${element.name}', '${element.name}_old_shape'],
     		    shape=(0,${element.element.outputTypes[0].channels?c},${element.element.outputTypes[0].height?c},${element.element.outputTypes[0].width?c}))
 </#if>
diff --git a/src/main/resources/templates/caffe2/elements/Output.ftl b/src/main/resources/templates/caffe2/elements/Output.ftl
index 26b28ebd9eeaf5064bab3a64e82ba808ba56df2e..ac35b057190658d38309904c09adaf36bca5b99f 100644
--- a/src/main/resources/templates/caffe2/elements/Output.ftl
+++ b/src/main/resources/templates/caffe2/elements/Output.ftl
@@ -4,8 +4,8 @@
 <#elseif element.logisticRegressionOutput>
     		${element.name} = model.net.Sigmoid(${input}, '${element.name}')
 <#elseif element.linearRegressionOutput>
-    		${element.name} = mx.symbol.LinearRegressionOutput(data=${element.inputs[0]}, <#-- TODO: check how to adapt linearRegressionOutput -->
-    		    name="${element.name}")
+    		<#--Don't add L2 loss here but within the function "add_training_operators" from CNNCreator.ftl-->
+    		${element.name} = ${input}
 </#if>
 
     		return ${element.name}
\ No newline at end of file