Skip to content

Commit a3f44c8

Browse files
committed
Test less for EfficientNet too
1 parent 7a32bb0 commit a3f44c8

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/convnets/efficientnet.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ Create an EfficientNet model ([reference](https://arxiv.org/abs/1905.11946v5)).
1313
- `k`: kernel size
1414
- `s`: kernel stride
1515
- `e`: expansion ratio
16-
- `i`: block input channels
16+
- `i`: block input channels (will be scaled by global width scaling)
1717
- `o`: block output channels (will be scaled by global width scaling)
1818
- `inchannels`: number of input channels
1919
- `nclasses`: number of output classes
@@ -110,7 +110,7 @@ See also [`efficientnet`](#).
110110
- `k`: kernel size
111111
- `s`: kernel stride
112112
- `e`: expansion ratio
113-
- `i`: block input channels
113+
- `i`: block input channels (will be scaled by global width scaling)
114114
- `o`: block output channels (will be scaled by global width scaling)
115115
- `inchannels`: number of input channels
116116
- `nclasses`: number of output classes

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)