Commit 7195855
committed
switch to ReverseDiffVJP(false) for VJP calculations
ReverseDiffVJP(true) assumed there was no programmatic branching in the function
This is true for most simulations when the rate coefficients are constant, but the assumption breaks down when the rate coefficients aren't constant
Originally ReverseDiffVJP(false) gave NaNs when run and TrackerVJP() would crash (as it does currently) and so ReverseDiffVJP(true) was set as the default
This should resolve issues with inconsistency between forward sensitivities and adjoint sensitivities1 parent 94b834a commit 7195855
1 file changed
Lines changed: 1 addition & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
138 | 138 | | |
139 | 139 | | |
140 | 140 | | |
141 | | - | |
| 141 | + | |
142 | 142 | | |
143 | 143 | | |
144 | 144 | | |
| |||
0 commit comments