publish 0.1.2

tnbar · Apr 5, 2021 · 3db0934 · 3db0934
1 parent 5e6a6d0
commit 3db0934
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/tednet/__init__.py b/tednet/__init__.py
@@ -1,7 +1,7 @@
 # -*- coding: UTF-8 -*-
 
 __author__ = "Perry"
-__version__ = "0.1.1"
+__version__ = "0.1.2"
 
 from ._ops import *
 

diff --git a/tednet/tnn/tn_rnn.py b/tednet/tnn/tn_rnn.py
@@ -62,9 +62,6 @@ def forward(self, inputs: torch.Tensor, state: LSTMState):
         gate_ih = self.weight_ih(inputs)
         gate_ih = self.dropout_ih(gate_ih)
 
-        print(hx.shape)
-        print(self.weight_hh.shape)
-
         gate_hh = torch.mm(self.dropout_hh(hx), self.weight_hh.t())
         gates = gate_ih + gate_hh + self.bias_hh