search

DENG-MIT / CRNN.jl_old

Chemical Reaction Neural Network in Julia
1 stars 0 forks source link

Compute gradient #1

Closed jiweiqi closed 3 years ago

jiweiqi commented 4 years ago

https://diffeqflux.sciml.ai/dev/controlling_AD/#

jiweiqi commented 4 years ago

Useful introduction https://www.dianacai.com/blog/2020/04/13/machine-learning-julia-flux-autodiff/

Use pullback to return both loss and its gradient https://fluxml.ai/Zygote.jl/dev/adjoints/