Update PytorchToMsnhnet.py

msnh2012 · Sep 1, 2020 · 4253b63 · 4253b63
1 parent 1bdf54b
commit 4253b63
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/pytorch2Msnhnet/PytorchToMsnhnet.py b/tools/pytorch2Msnhnet/PytorchToMsnhnet.py
@@ -274,7 +274,7 @@ def _relu6(raw, inData, inplace=False):
     return x
 
 def _leaky_relu(raw, inData, negative_slope=0.01, inplace=False):
-    log( "leaky-i" , args[0]._cdata)
+    log( "leaky-i" , inData._cdata)
     x = raw(inData, negative_slope,inplace)
     ccc.append(x)
     log( "leaky-o" , x._cdata)
@@ -702,7 +702,7 @@ def _log10(raw, inData, *args):
     return x
 
 def _contiguous(inData, *args):
-    log( "contiguous-i" , args[0]._cdata)
+    log( "contiguous-i" , inData._cdata)
     x = raw__contiguous__(inData, *args)
     ccc.append(x)