bug fix in HED example

examples/HED/hed.py

@@ -71,7 +71,7 @@ class Model(ModelDesc):
         for idx, b in enumerate([b1, b2, b3, b4, b5, final_map]):
             output = tf.nn.sigmoid(b, name='output{}'.format(idx+1))
             xentropy = class_balanced_sigmoid_cross_entropy(
-                b, edgemap,
+                tf.squeeze(b, [3]), edgemap,
                 name='xentropy{}'.format(idx+1))
             costs.append(xentropy)