wangkuiyi/gotorch

Support PyTorch state_dict

Closed this issue · 1 comments

wangkuiyi commented

As described https://pytorch.org/tutorials/beginner/saving_loading_models.html#saving-loading-model-for-inference

It seems that

  1. We should rename Module.GetNamedParameters into Module.StateDict.
  2. We should add a Module.LoadStateDict.
wangkuiyi commented

Fixed by #247 and #250