Upgrade same/his

2021-04-29 04:48:21 -07:00
parent f7c2bb5e32
commit 1209fffbaa
4 changed files with 90 additions and 5 deletions
--- a/exps/LFNA/basic-his.py
+++ b/exps/LFNA/basic-his.py
@@ -136,7 +136,8 @@ def main(args):
        )
        save_checkpoint(
            {
-                "model": model.state_dict(),
+                "model_state_dict": model.state_dict(),
+                "model": model,
                "index": idx,
                "timestamp": env_info["{:}-timestamp".format(idx)],
            },