What is the recommended approach to exclude parts of my loss functiin from being targeted by Zygote? I am using Lux.jl. Thanks.
erlebach
1
ChainRulesCore.ignore_derivatives
https://juliadiff.org/ChainRulesCore.jl/stable/api.html#Ignoring-gradients
Related topics
| Topic | Replies | Views | Activity | |
|---|---|---|---|---|
| Ignore derivatives in ReverseDiff | 0 | 395 | April 8, 2023 | |
| Ignore part of the gradient calculation inside differential equation model | 0 | 316 | November 9, 2022 | |
| Avoid storing intermediate results from the forward pass by default | 14 | 381 | September 29, 2024 | |
| Zygote and TimerOutputs, derivative of timed function | 3 | 645 | March 7, 2022 | |
| Ignore_derivatives of entire module with ChainRules.jl | 1 | 391 | February 28, 2023 |