LuxDL/Lux.jl

Preserve `init` functions for SimpleChains

開放

#582 建立於 2024年4月10日

 (0 則留言) (0 個反應) (0 位負責人)Julia (84 個分叉)batch import
enhancementgood first issue

倉庫指標

星標
 (700 顆星)
PR 合併指標
 (平均合併 13天 21小時) (30 天內合併 2 個 PR)

描述

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.

貢獻者指南