Skip to content

Commit 0d2b842

Browse files
committed
Fix missing constructor arg for Xception
1 parent 327b345 commit 0d2b842

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/convnets/inception.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -579,8 +579,9 @@ Creates an Xception model.
579579
580580
`Xception` does not currently support pretrained weights.
581581
"""
582-
function Xception(; inchannels = 3, dropout = 0.0, nclasses = 1000)
582+
function Xception(; pretrain = false, inchannels = 3, dropout = 0.0, nclasses = 1000)
583583
layers = xception(; inchannels, dropout, nclasses)
584+
pretrain && loadpretrain!(layers, "xception")
584585
return Xception(layers)
585586
end
586587

0 commit comments

Comments
 (0)