Skip to content

Commit aa2a9ef

Browse files
committed
Tests just don't fix themselves
1 parent 674b27e commit aa2a9ef

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

test/convnets.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ end
107107
end
108108

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

test/mixers.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
@testset "MLPMixer" begin
2-
@testset for mode in [:small, :base, :large, :huge]
2+
@testset for mode in [:small, :base, :large] #:huge]
33
@testset for drop_path_rate in [0.0, 0.5]
44
m = MLPMixer(mode; drop_path_rate)
55
@test size(m(x_224)) == (1000, 1)
@@ -10,7 +10,7 @@
1010
end
1111

1212
@testset "ResMLP" begin
13-
@testset for mode in [:small, :base, :large, :huge]
13+
@testset for mode in [:small, :base, :large] #:huge]
1414
@testset for drop_path_rate in [0.0, 0.5]
1515
m = ResMLP(mode; drop_path_rate)
1616
@test size(m(x_224)) == (1000, 1)
@@ -21,7 +21,7 @@ end
2121
end
2222

2323
@testset "gMLP" begin
24-
@testset for mode in [:small, :base, :large, :huge]
24+
@testset for mode in [:small, :base, :large] #:huge]
2525
@testset for drop_path_rate in [0.0, 0.5]
2626
m = gMLP(mode; drop_path_rate)
2727
@test size(m(x_224)) == (1000, 1)

0 commit comments

Comments
 (0)