Use acctest for efficient net

FluxML · Jun 19, 2022 · 2242785 · 2242785
1 parent 4f86b62
commit 2242785
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/convnets.jl b/test/convnets.jl
@@ -77,7 +77,7 @@ GC.gc()
         m = EfficientNet(name)
         @test size(m(x)) == (1000, 1)
         if (EfficientNet, name) in PRETRAINED_MODELS
-            @test (EfficientNet(name, pretrain = true); true)
+            @test acctest(EfficientNet(name, pretrain = true))
         else
             @test_throws ArgumentError EfficientNet(name, pretrain = true)
         end