nilboy / tensorflow-yolo

tensorflow implementation of 'YOLO : Real-Time Object Detection'(train and test)
774 stars 312 forks source link

what does the weight_loss mean in loss function? #52

Open ZhouJiaHuan opened 6 years ago

contactmat85 commented 6 years ago

I didn't understand it neither

mangwaier commented 5 years ago
  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.