Skip to content

Commit 01ab049

Browse files
committed
Only test smaller variants on efficient net
1 parent aa1d37d commit 01ab049

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/convnets.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ GC.safepoint()
7171
GC.gc()
7272

7373
@testset "EfficientNet" begin
74-
@testset "EfficientNet($name)" for name in [:b0, :b1, :b2, :b3, :b4, :b5, :b6, :b7, :b8]
74+
@testset "EfficientNet($name)" for name in [:b0, :b1, :b2] #, :b3, :b4, :b5, :b6, :b7, :b8]
7575
# preferred image resolution scaling
7676
r = Metalhead.efficientnet_global_configs[name][1]
7777
x = rand(Float32, r, r, 3, 1)

0 commit comments

Comments
 (0)