The solution to this is the same as The tracked parameters are not updated during Flux.training - #2 by ChrisRackauckas
1 Like
The solution to this is the same as The tracked parameters are not updated during Flux.training - #2 by ChrisRackauckas