Skip to content

Commit 61d886b

Browse files
committed
Fix size error in test setup
1 parent 19a18bf commit 61d886b

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

test/convnets.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,8 +72,9 @@ GC.gc()
7272

7373
@testset "EfficientNet" begin
7474
@testset "EfficientNet($name)" for name in [:b0, :b1, :b2, :b3, :b4, :b5, :b6, :b7, :b8]
75-
xsz = Metalhead.efficientnet_global_configs[name][1]
76-
x = rand(Float32, xsz...)
75+
# preferred image resolution scaling
76+
r = Metalhead.efficientnet_global_configs[name][1]
77+
x = rand(Float32, r, r, 3, 1)
7778
m = EfficientNet(name)
7879
@test size(m(x)) == (1000, 1)
7980
if (EfficientNet, name) in PRETRAINED_MODELS

0 commit comments

Comments
 (0)