search

apache / singa

a distributed deep learning platform
Apache License 2.0
3.35k stars 1.24k forks source link

Need to return the gradients from optimizer #1081

Closed daoducanhc closed 1 year ago

daoducanhc commented 1 year ago

Need to return the gradients from the optimizer for calculating the training-free metric

daoducanhc commented 1 year ago

the gradients are returned in the line 242 of https://github.com/apache/singa/commit/d9831d917ae0918749faacf5a611b161cbdd1fc9#diff-882c3945309d2e5184aa4c29cbbcc2eb0c95911af8319b5332d049ba72f55aecR242

zmeihui commented 1 year ago

In Line 432 of https://github.com/apache/singa/blob/dev-postgresql/examples/model_selection_psql/ms_mlp/train_mlp.py#L432

the gradients are returned and used in the training process

daoducanhc commented 1 year ago

resolved in https://github.com/apache/singa/commit/d9831d917ae0918749faacf5a611b161cbdd1fc9

daoducanhc commented 1 year ago

resolved in https://github.com/apache/singa/commit/d9831d917ae0918749faacf5a611b161cbdd1fc9