search

cedrickchee / capsule-net-pytorch

[NO MAINTENANCE INTENDED] A PyTorch implementation of CapsNet architecture in the NIPS 2017 paper "Dynamic Routing Between Capsules".
Other
168 stars 50 forks source link

Show training log for reconstruction loss #2

Closed cedrickchee closed 6 years ago

cedrickchee commented 6 years ago

Currently, the training log only shows the total loss.

For the sake of debugging and monitoring the learning progress of the network, we should show both the margin and reconstruction loss separately.

cedrickchee commented 6 years ago

Implemented in this commit: 66b65e044c19680cb76a31dad6cc983b4aed6375.

We are showing these losses both in TensorBoard plots and terminal/command line during training and testing:

  1. Total loss
  2. Margin loss
  3. Reconstruction loss

We need to update the README to document down and explain this change.