LuxDL/Lux.jl

Preserve `init` functions for SimpleChains

Aperta

#582 aperta il 10 apr 2024

 (0 commenti) (0 reazioni) (0 assegnatari)Julia (84 fork)batch import
enhancementgood first issue

Metriche repository

Star
 (700 stelle)
Metriche merge PR
 (Merge medio 13g 21h) (2 PR mergiate in 30 g)

Descrizione

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.

Guida contributor