vietnh1009 / Yolo-v2-pytorch

YOLO for object detection tasks
MIT License
373 stars 107 forks source link

Convolution in last layer should have bias = True #8

Open HamsterHuey opened 5 years ago

HamsterHuey commented 5 years ago

All other Conv2D layers have bias=False since they are followed by a batch-norm layer which essentially has a bias term. However, the final layer does not have batch-norm at the output and should have a bias term. This is how it is in the reference YOLO implementation.

https://github.com/vietnguyen91/Yolo-v2-pytorch/blob/9589413b5dce0476eb9cccc41945cf30cf131b34/src/yolo_net.py#L64