Co-authored-by: rudongyu <ru_dongyu@outlook.com>
state_dict
load_state_dict
param_groups
dgl.optim.SparseGradOptimizer