MCPcopy Create free account
hub / github.com/InnerPeace-Wu/CapsNet-tensorflow / _add_loss

Method _add_loss

code/CapsNet.py:220–281  ·  view source on GitHub ↗

add the margin loss and reconstruction loss :arg digit_caps: output of digit capsule layer, shape [None, 10, 16] :return total_loss:

(self)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

train_architectureMethod · 0.95

Calls 1

_reconstructMethod · 0.95

Tested by

no test coverage detected