search

naturomics / CapsNet-Tensorflow

A Tensorflow implementation of CapsNet(Capsules Net) in paper Dynamic Routing Between Capsules
Apache License 2.0
3.8k stars 1.17k forks source link

mistake on annotation #13

Closed jihobak closed 6 years ago

jihobak commented 6 years ago

I think there are some mistake on annotation. I found two.

L61 ` Reshape the input into [batch_size, 1, 1152, 8, 1] --->

Reshape the input into [batch_size, 1152, 1, 8, 1] `

L77 ` input: A Tensor with [batch_size, 1, num_caps_l=1152, length(u_i)=8, 1] --->

input: A Tensor with [batch_size, num_caps_l=1152, 1, length(u_i)=8, 1] `

order should be change I think

Thanks

naturomics commented 6 years ago

merged