tensorflow-yolo icon indicating copy to clipboard operation
tensorflow-yolo copied to clipboard

what does the weight_loss mean in loss function?

Open ZhouJiaHuan opened this issue 6 years ago • 2 comments

ZhouJiaHuan avatar Mar 30 '18 03:03 ZhouJiaHuan

I didn't understand it neither

contactmat85 avatar Aug 17 '18 08:08 contactmat85

  weight_decay = tf.multiply(tf.nn.l2_loss(var), wd, name='weight_loss')
  tf.add_to_collection('losses', weight_decay)

As the code say, there is a weight decay based on L2.

mangwaier avatar Jan 29 '19 08:01 mangwaier