Skip to content

Commit 13bb641

Browse files
committed
remove dead code
1 parent 0ae4f97 commit 13bb641

File tree

1 file changed

+0
-8
lines changed

1 file changed

+0
-8
lines changed

src/methods/imageclassification.jl

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -169,14 +169,6 @@ function DLPipelines.methodmodel(method::ImageClassification, backbone)
169169
h, w, ch, b = Flux.outdims(backbone, (method.projections.sz..., 3, 1))
170170
head = Models.visionhead(ch, length(method.classes), p = 0.)
171171
return Chain(backbone, head)
172-
return Chain(
173-
backbone,
174-
Chain(
175-
AdaptiveMeanPool((1, 1)),
176-
flatten,
177-
Dense(ch, length(method.classes)),
178-
)
179-
)
180172
end
181173

182174
DLPipelines.methodlossfn(::ImageClassification) = Flux.Losses.logitcrossentropy

0 commit comments

Comments
 (0)