Skip to content

Commit a83efd9

Browse files
committed
No need to use image resolution scaling (implicitly done)
1 parent d7e2e19 commit a83efd9

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

src/convnets/efficientnet.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@ function EfficientNet(name::Symbol; pretrain = false)
147147
@assert name in keys(efficientnet_global_configs)
148148
"`name` must be one of $(sort(collect(keys(efficientnet_global_configs))))"
149149

150-
model = EfficientNet(efficientnet_global_configs[name]..., efficientnet_block_configs)
150+
model = EfficientNet(efficientnet_global_configs[name][2], efficientnet_block_configs)
151151
pretrain && loadpretrain!(model, string("efficientnet-", name))
152152

153153
return model

test/convnets.jl

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,14 +72,16 @@ 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...)
7577
m = EfficientNet(name)
76-
@test size(m(x_256)) == (1000, 1)
78+
@test size(m(x)) == (1000, 1)
7779
if (EfficientNet, name) in PRETRAINED_MODELS
7880
@test (EfficientNet(name, pretrain = true); true)
7981
else
8082
@test_throws ArgumentError EfficientNet(name, pretrain = true)
8183
end
82-
@test gradtest(m, x_256)
84+
@test gradtest(m, x)
8385
GC.safepoint()
8486
GC.gc()
8587
end

0 commit comments

Comments
 (0)