From 7aff0e3a400eddbc8d25bfe0d4ba019ef073c3b4 Mon Sep 17 00:00:00 2001 From: Kyle Daruwalla Date: Sun, 19 Jun 2022 10:12:08 +0530 Subject: [PATCH] Fix missing constructor arg for Xception --- src/convnets/inception.jl | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) 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