norm(Δ)
This commit is contained in:
parent
7d6f711c6f
commit
114f63a214
|
@ -555,7 +555,7 @@ mutable struct ClipNorm{T}
|
||||||
end
|
end
|
||||||
|
|
||||||
function apply!(o::ClipNorm, x, Δ)
|
function apply!(o::ClipNorm, x, Δ)
|
||||||
Δnrm = norm(Δ, 2)
|
Δnrm = norm(Δ)
|
||||||
if Δnrm > o.thresh
|
if Δnrm > o.thresh
|
||||||
rmul!(Δ, o.thresh / Δnrm)
|
rmul!(Δ, o.thresh / Δnrm)
|
||||||
end
|
end
|
||||||
|
|
Loading…
Reference in New Issue