diff --git a/capsLayer.py b/capsLayer.py index 3599431..b649b3e 100644 --- a/capsLayer.py +++ b/capsLayer.py @@ -81,7 +81,7 @@ def capsule(input, b_IJ, idx_j): Notes: u_i represents the vector output of capsule i in the layer l, and v_j the vector output of capsule j in the layer l+1. - ''' + ''' with tf.variable_scope('routing'): w_initializer = np.random.normal(size=[1, 1152, 8, 16], scale=0.01) diff --git a/capsNet.py b/capsNet.py index 1cfae25..82eb4be 100644 --- a/capsNet.py +++ b/capsNet.py @@ -35,7 +35,7 @@ def build_arch(self): assert conv1.get_shape() == [cfg.batch_size, 20, 20, 256] # TODO: Rewrite the 'CapsConv' class as a function, the capsLay - # function should be encapsulated into tow function, one like conv2d + # function should be encapsulated into two functions, one like conv2d # and another is fully_connected in Tensorflow. # Primary Capsules, [batch_size, 1152, 8, 1] with tf.variable_scope('PrimaryCaps_layer'):