diff --git a/mindspore/nn/cell.py b/mindspore/nn/cell.py index e30daab52396d184d9cd3137efc01af80eaab1b6..888aa08ac7b16e3855412f9595b9c5dcad312707 100755 --- a/mindspore/nn/cell.py +++ b/mindspore/nn/cell.py @@ -103,6 +103,16 @@ class Cell(Cell_): def already_run(self): return self._already_run + def __getstate__(self): + base = Cell_.__getstate__(self) + return base, self.__dict__ + + def __setstate__(self, state): + base, dict_ = state + Cell_.__setstate__(self, base) + self.__dict__ = dict_ + self._attr_synced = False + @property def _cell_tag(self): # `` @@ -388,6 +398,9 @@ class Cell(Cell_): """ return '' + def __str__(self): + return self.__repr__() + def __repr__(self): extra_str = self.extend_repr() info_str = self.__class__.__name__ + '<'