Capsule-Network-Tutorial
Capsule-Network-Tutorial copied to clipboard
margin loss wrong
Hello, I think your magin loss is wrong, according to the paper, after relu, there is a square you forgot
I think it should be left = ((F.relu(0.9 - v_c))**2).view(batch_size, -1) right = ((F.relu(v_c - 0.1))**2).view(batch_size, -1)
I have also noticed that little mistake and reviewing the paper it is clear that there are two missing squares.