Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion capsLayer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
2 changes: 1 addition & 1 deletion capsNet.py
Original file line number Diff line number Diff line change
Expand Up @@ -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'):
Expand Down