update

DoodleJZ · Aug 19, 2019 · 8d5bdf9 · 8d5bdf9
1 parent 9792e2c
commit 8d5bdf9
Show file tree

Hide file tree

Showing 3 changed files with 61 additions and 40 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/src_division/Zparser.py b/src_division/Zparser.py
@@ -83,7 +83,7 @@ def __init__(self, batch_idxs_np):
         self.max_len = int(np.max(self.boundaries_np[1:] - self.boundaries_np[:-1]))
 
 #
-class FeatureDropoutFunction(nn.functional._functions.dropout.InplaceFunction):
+class FeatureDropoutFunction(torch.autograd.function.InplaceFunction):
     @classmethod
     def forward(cls, ctx, input, batch_idxs, p=0.5, train=False, inplace=False):
         if p < 0 or p > 1:

diff --git a/src_joint/Zparser.py b/src_joint/Zparser.py
@@ -85,7 +85,7 @@ def __init__(self, batch_idxs_np):
         self.max_len = int(np.max(self.boundaries_np[1:] - self.boundaries_np[:-1]))
 
 #
-class FeatureDropoutFunction(nn.functional._functions.dropout.InplaceFunction):
+class FeatureDropoutFunction(torch.autograd.function.InplaceFunction):
     @classmethod
     def forward(cls, ctx, input, batch_idxs, p=0.5, train=False, inplace=False):
         if p < 0 or p > 1: