Yolo-v2-pytorch icon indicating copy to clipboard operation
Yolo-v2-pytorch copied to clipboard

Convolution in last layer should have bias = True

Open HamsterHuey opened this issue 5 years ago • 0 comments

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

HamsterHuey avatar May 04 '19 18:05 HamsterHuey