dmlc--dgl

Author SHA1 Message Date

Author	SHA1	Message	Date
Xin Yao	ae4a5b7394	[Feature] Add `state_dict`, `load_state_dict`, `param_groups` to `dgl.optim.SparseGradOptimizer` (#5311 ) * init update * all get/set optm_state * add unit tests * add docstring * fix for multiple embeddings * move embedding methods to private * fix lint * fix unit tests * resolve comments * merge master	2023-03-08 16:46:34 +08:00
Hongzhi (Steve), Chen	86193c26f7	[Misc] Reorg python test (with backend) according to the guideline. (#5312 ) * rename * remove test transpose * udpate * rename * cleanup --------- Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>	2023-02-17 16:11:39 +08:00

Xin Yao

ae4a5b7394

[Feature] Add state_dict, load_state_dict, param_groups to dgl.optim.SparseGradOptimizer (#5311 )

* init update

* all get/set optm_state

* add unit tests

* add docstring

* fix for multiple embeddings

* move embedding methods to private

* fix lint

* fix unit tests

* resolve comments

* merge master

2023-03-08 16:46:34 +08:00

Hongzhi (Steve), Chen

86193c26f7

[Misc] Reorg python test (with backend) according to the guideline. (#5312 )

* rename

* remove test transpose

* udpate

* rename

* cleanup

---------

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>

2023-02-17 16:11:39 +08:00

2 Commits