Skip to content

Commit 1cd9a6e

Browse files
committed
Minor fix
1 parent 3df322d commit 1cd9a6e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/src/models.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,7 @@ end
202202

203203
function (bn::BottleneckBlock)((x, y)::Tuple{<:AbstractArray,<:AbstractArray}, ps, st)
204204
x_r, st_rescale = bn.rescale(x, ps.rescale, st.rescale)
205-
x_m, st_conv1 = bn.conv(x_r, ps.conv, st.conv)
205+
x_m, st_conv1 = bn.conv(x, ps.conv, st.conv)
206206

207207
x_m = y .+ x_m
208208
x_m, st_mapping = bn.mapping(x_m, ps.mapping, st.mapping)
@@ -219,7 +219,7 @@ end
219219

220220
function (bn::BottleneckBlock)(x::AbstractArray, ps, st)
221221
x_r, st_rescale = bn.rescale(x, ps.rescale, st.rescale)
222-
x_m, st_conv1 = bn.conv(x_r, ps.conv, st.conv)
222+
x_m, st_conv1 = bn.conv(x, ps.conv, st.conv)
223223
x_m, st_mapping = bn.mapping(x_m, ps.mapping, st.mapping)
224224

225225
return (

0 commit comments

Comments
 (0)