Hey, how can I clip my weights using tracked parameters ?
I know I can use something like
Zygote.gradient(Tracker.data.(params(m))) do x
total_loss(Zygote.hook(ā -> ā./norm(a),x))
end
However, I use Flux.mse for my loss whose arguments aren’t the weights. How can I proceed?