Skip to content

Commit

Permalink
Update src/denoising_model.jl
Browse files Browse the repository at this point in the history
  • Loading branch information
dpo authored Feb 20, 2024
1 parent 47a2939 commit a168a30
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/denoising_model.jl
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ function denoising_model(shape, shape_p, KERNEL_SIZE, KERNEL_SIGMA = 1.5)

function grad!(g, x)
y .= H(W_T(x))
z .= 1.0 ./ ((y .- b) .^ 2 .+ 1)
z .= 1 ./ ((y .- b) .^ 2 .+ 1)
@. z = 2 * z * (y - b)
g .= W(H_T(z))
return g
Expand Down

0 comments on commit a168a30

Please sign in to comment.