From 470e490bd1565f6afbf623bab952c7eebdc0a243 Mon Sep 17 00:00:00 2001 From: Yang Zhang Date: Tue, 7 Jan 2020 17:01:44 +0800 Subject: [PATCH] Save optimizer only when it has state --- model.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/model.py b/model.py index 2e53bd1..a2ae5d0 100644 --- a/model.py +++ b/model.py @@ -178,6 +178,8 @@ class StaticGraphAdapter(object): optim_path = path + ".pdopt" optim = {p.name: p for p in filter( is_belong_to_optimizer, prog.list_vars())} + if not optim: + return # HACK this is contrived, optimizer state is not the same for # static/dynamic graph mode optim['__static_graph_only__'] = True -- GitLab