Skip to content
This repository was archived by the owner on Apr 23, 2025. It is now read-only.

Commit 87da673

Browse files
committed
Mini-cleanup of ResNet
1 parent 7bb3e56 commit 87da673

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

Models/ImageClassification/ResNet.swift

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -47,10 +47,8 @@ public func autoResidualBlock(inputFilters: Int, filters: Int, strides: (Int, In
4747
var earlyConvs: [AutoConvBN] = []
4848
let lastConv: AutoConvBN
4949
if isBasic {
50-
earlyConvs = [
51-
(autoConvBN(
52-
filterShape: (3, 3), outputChannels: filters, strides: strides, padding: .same)),
53-
]
50+
earlyConvs.append(
51+
autoConvBN(filterShape: (3, 3), outputChannels: filters, strides: strides, padding: .same))
5452
lastConv = autoConvBN(filterShape: (3, 3), outputChannels: outFilters, padding: .same)
5553
} else {
5654
if useLaterStride {

0 commit comments

Comments
 (0)