diff --git a/src/convnets/inception.jl b/src/convnets/inception.jl index ceedf7aa7..ebd7ab809 100644 --- a/src/convnets/inception.jl +++ b/src/convnets/inception.jl @@ -579,9 +579,10 @@ Creates an Xception model. `Xception` does not currently support pretrained weights. """ -function Xception(; inchannels = 3, dropout = 0.0, nclasses = 1000) - layers = xception(; inchannels, dropout, nclasses) - return Xception(layers) +function Xception(; pretrain = false, inchannels = 3, dropout = 0.0, nclasses = 1000) + layers = Xception(xception(; inchannels, dropout, nclasses)) + pretrain && loadpretrain!(layers, "xception") + return layers end @functor Xception