ProgJ 2018-03-27 12:07 采纳率: 33.3%
浏览 1088
已结题

tensorflow capsule胶囊算法的loss是如何定义的

我在git找的代码是这样定义的:
# 3. Total loss
# The paper uses sum of squared error as reconstruction error, but we
# have used reduce_mean in # 2 The reconstruction loss to calculate
# mean squared error. In order to keep in line with the paper,the
# regularization scale should be 0.0005*784=0.392
self.total_loss = self.margin_loss + cfg.regularization_scale * self.reconstruction_err

各位有没有详细一点的解释,它这里好像还把重构(reconstruction)的loss加进去了?但是好像看网上资料里面loss只有margin loss?

  • 写回答

1条回答 默认 最新

报告相同问题?