Lux.jl icon indicating copy to clipboard operation
Lux.jl copied to clipboard

Preserve `init` functions for SimpleChains

Open avik-pal opened this issue 1 year ago • 0 comments
trafficstars

Currently we drop all of the init functions from the Lux models. However, for SciML Applications it is quite critical to support those. Proposed solution: Store the lux model along side the equivalent simple chains version. Instead of calling params on the simplechains model call initialparameters on the lux model and then flatten it.

avik-pal avatar Apr 10 '24 22:04 avik-pal