diff --git a/tensorpack/coupled.py b/tensorpack/coupled.py index c4ad3f8..054200f 100644 --- a/tensorpack/coupled.py +++ b/tensorpack/coupled.py @@ -62,7 +62,7 @@ def initialize(self, method="svd", verbose=False): # wipe off old values self.x["_Weight_"][:] = np.ones_like(self.x["_Weight_"]) for mmode in self.modes: - self.x["_" + mmode][:] = np.zeros_like(self.x["_" + mmode]) + self.x["_" + mmode][:] = np.ones_like(self.x["_" + mmode]) if method == "ones": for mmode in self.modes: