Capsule-Network-Tutorial icon indicating copy to clipboard operation
Capsule-Network-Tutorial copied to clipboard

Published 20 hours ago • verified publisher icon higgsfield

Change num_capusle to num_conv2D_per_capsule

Open himalayajung opened this issue 6 years ago • 0 comments

Change the input variable num_capusle to num_conv2D_per_capsule in the class PrimaryCaps because 8 is not the number of capsules, it is the number of conv2D units in one primary capsule. From the paper--"each primary capsule contains 8 convolutional units with a 9×9 kernel and a stride of 2"

himalayajung avatar Mar 29 '18 16:03 himalayajung