diff --git a/JayRnn.py b/JayRnn.py index 8748946..ab7ab75 100644 --- a/JayRnn.py +++ b/JayRnn.py @@ -82,16 +82,13 @@ class RnnModel(Model): self.units.append(unit) def train(self, vc: Mat, status: Status = None): - for i, unit in enumerate(self.units): + for unit in self.units: train(unit, vc, status) # Update context portion of vc vc[:, WIN*vocab_size():] = unit.forward(vc) def forward_step(self, v_curr: Mat): - c = np.zeros([1, H_SIZE]) - z = np.zeros(v_curr.shape) - for i, unit in enumerate(self.units): - pass + pass def forward(self, x: Mat): pass