fix context nonetype bug

4 years ago · 2172454997
parent db89b1674c
commit 2172454997
1 changed files with 1 additions and 1 deletions
--- a/mindspore/nn/cell.py
+++ b/mindspore/nn/cell.py
@ -244,7 +244,7 @@ class Cell(Cell_):
        raise AttributeError("'{}' object has no attribute '{}'.".format(type(self).__name__, name))

    def __del__(self):
-        if context.get_context("mode") == context.PYNATIVE_MODE:
+        if context is not None and context.get_context("mode") == context.PYNATIVE_MODE:
            _pynative_exec.clear(str(id(self)))
        if hasattr(self, "_create_time"):
            _executor.del_net_res(str(self._create_time))