This commit is contained in:
Glenn Jocher committed 2020-06-02 23:40:14 -07:00
1 parent 11b462e2a9
commit 17e4926cdf
2 files changed
+3 -2

No files matched your search

+1 -1
View File
@@ -26,7 +26,7 @@ backbone:
# yolov5 head
head:
[[-1, 3, Bottleneck, [1024]], # 11
[[-1, 3, Bottleneck, [1024, False]], # 11
[-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], # 12 (P5/32-large)
[-2, 1, nn.Upsample, [None, 2, 'nearest']],