提交 8d975e6d authored 作者: sentient07's avatar sentient07

Moved the variables into on_attach method

上级 9625e083
...@@ -266,8 +266,6 @@ class ReplaceValidate(History, Validator): ...@@ -266,8 +266,6 @@ class ReplaceValidate(History, Validator):
pickle_rm_attr = (["replace_validate", "replace_all_validate", pickle_rm_attr = (["replace_validate", "replace_all_validate",
"replace_all_validate_remove"] + "replace_all_validate_remove"] +
History.pickle_rm_attr + Validator.pickle_rm_attr) History.pickle_rm_attr + Validator.pickle_rm_attr)
_nodes_removed = set()
fail_validate = False
def on_attach(self, fgraph): def on_attach(self, fgraph):
for attr in ('replace_validate', 'replace_all_validate', for attr in ('replace_validate', 'replace_all_validate',
...@@ -275,6 +273,8 @@ class ReplaceValidate(History, Validator): ...@@ -275,6 +273,8 @@ class ReplaceValidate(History, Validator):
if hasattr(fgraph, attr): if hasattr(fgraph, attr):
raise AlreadyThere("ReplaceValidate feature is already present" raise AlreadyThere("ReplaceValidate feature is already present"
" or in conflict with another plugin.") " or in conflict with another plugin.")
self._nodes_removed = set()
self.fail_validate = False
History.on_attach(self, fgraph) History.on_attach(self, fgraph)
Validator.on_attach(self, fgraph) Validator.on_attach(self, fgraph)
self.unpickle(fgraph) self.unpickle(fgraph)
...@@ -377,8 +377,6 @@ class ReplaceValidate(History, Validator): ...@@ -377,8 +377,6 @@ class ReplaceValidate(History, Validator):
self.fail_validate = True self.fail_validate = True
def validate(self, fgraph): def validate(self, fgraph):
if not hasattr(fgraph, 'destroyers'):
return True
if self.fail_validate: if self.fail_validate:
self.fail_validate = False self.fail_validate = False
raise theano.gof.InconsistencyError("Trying to reintroduce a removed node") raise theano.gof.InconsistencyError("Trying to reintroduce a removed node")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论