Further RC2 update

cjsure · Mar 24, 2015 · 18c7952 · 18c7952
1 parent 829f439
commit 18c7952
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/CaffeGenerate.py b/CaffeGenerate.py
@@ -14,7 +14,7 @@ def convtemplate(name, OutputLs, Padding, kernelsize, Stride, bottom, bfv, flr,
     string = \
         'layer {\n\
         name: "%s"\n\
-        type: "convolution"\n\
+        type: "Convolution"\n\
         param {\n\
         lr_mult: %i\n\
         decay_mult: %i\n\
@@ -220,7 +220,7 @@ def SMtemplate(name, bottom, w):
     string = \
         'layer {\n\
         name: "loss"\n\
-        type: "Softmax"\n\
+        type: "SoftmaxWithLoss"\n\
         bottom: "%s"\n\
         bottom: "label"\n\
         top: "%s"\n\