save()/load() tests and fixes
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/23911
Test Plan: Imported from OSS
Differential Revision: D16698044
Pulled By: jamesr66a
fbshipit-source-id: 88881ea183331aa6e4c8fa042d11cf2b14e0fc4c
