search

nnstreamer / nntrainer

NNtrainer is Software Framework for Training Neural Network Models on Devices.
Apache License 2.0
147 stars 73 forks source link

Adam variable loading semantics #1845

Open kparichay opened 2 years ago

kparichay commented 2 years ago

Adam variables are loaded with ml::train::model::load(), which allows loading the model weights. However, model weights can be loaded with a call to ml::train::model::load() but adam weights can only be loaded when load() is called from inside the train() as adam parameters are only available during the training. The error message in the former case needs to be set properly.

taos-ci commented 2 years ago

:octocat: cibot: Thank you for posting issue #1845. The person in charge will reply soon.