updates
This commit is contained in:
1 parent
11b462e2a9
commit
17e4926cdf
2 files changed
+3
-2
No files matched your search
+1
-1
@@ -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']],
|
||||
|
||||
Reference in new issue
Block a user