Adjust testing and use Chain(::Vector)

Author: darsnack

src/convnets/efficientnet.jl

@@ -53,7 +53,7 @@ function efficientnet(scalings, block_config;
 
     top = Dense(head_out_channels, nclasses)
 
-    return Chain(Chain(stem..., blocks, head...),
+    return Chain(Chain([stem..., blocks, head...]),
                  Chain(AdaptiveMeanPool((1, 1)), MLUtils.flatten, top))
 end
 

test/other.jl

@@ -1,5 +1,5 @@
 @testset "MLPMixer" begin
-	@testset for mode in [:small, :base, :large] # :huge]
+	@testset for mode in [:small, :base] # :large, # :huge]
 		@testset for drop_path_rate in [0.0, 0.5]
 			m = MLPMixer(mode; drop_path_rate)
 			@test size(m(x_224)) == (1000, 1)
@@ -11,7 +11,7 @@
 end
 
 @testset "ResMLP" begin
-  @testset for mode in [:small, :base, :large] # :huge]
+  @testset for mode in [:small, :base] # :large, # :huge]
 		@testset for drop_path_rate in [0.0, 0.5]
 			m = ResMLP(mode; drop_path_rate)
 			@test size(m(x_224)) == (1000, 1)
@@ -23,7 +23,7 @@ end
 end
 
 @testset "gMLP" begin
-    @testset for mode in [:small, :base, :large] # :huge]
+    @testset for mode in [:small, :base] # :large, # :huge]
 		@testset for drop_path_rate in [0.0, 0.5]
 			m = gMLP(mode; drop_path_rate)
 			@test size(m(x_224)) == (1000, 1)