Skip to content

Commit 7e2e1e4

Browse files
committed
Fix a copy paste error for _round_channels
1 parent 5e5a56e commit 7e2e1e4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/convnets/efficientnet.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,8 @@ function efficientnet(scalings, block_config;
2929

3030
blocks = []
3131
for (n, k, s, e, i, o) in block_config
32-
in_channels = round_filter(i, 8)
33-
out_channels = round_filter(wscale 1 ? o : ceil(Int64, wscale * o), 8)
32+
in_channels = _round_channels(i, 8)
33+
out_channels = _round_channels(wscale 1 ? o : ceil(Int64, wscale * o), 8)
3434
repeat = dscale 1 ? n : ceil(Int64, dscale * n)
3535

3636
push!(blocks,

0 commit comments

Comments
 (0)