diff --git a/ivy/stateful/module.py b/ivy/stateful/module.py index d8deaa031d989..c89e9706ad3d7 100644 --- a/ivy/stateful/module.py +++ b/ivy/stateful/module.py @@ -778,7 +778,7 @@ def _build(self, *args, **kwargs): import tensorflow as tf def _get_variable_name(variable): - return variable.path.split("/")[-2] + "/" + variable.name + return variable.path.split("/")[-2] + "/" + variable.name + ":0" self._native_params = ivy.Container( OrderedDict(