Skip to content

Commit

Permalink
Update dsconv2d_cplx.py
Browse files Browse the repository at this point in the history
  • Loading branch information
felixfuyihui authored Jun 29, 2022
1 parent 6ac30f8 commit 8e3fd61
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions dsconv2d_cplx.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 4,6 @@
import torch
import torch.nn as nn
import torch.nn.functional as F
import os
import sys

from conv2d_cplx import ComplexConv2d_Encoder
EPSILON = torch.finfo(torch.float32).eps
Expand Down Expand Up @@ -58,7 56,7 @@ def forward(self, x):
y = y * torch.sigmoid(y)
y = self.sconv(y)
y = self.dropout(y)
x = x y
y = x y
return y

if __name__ == '__main__':
Expand Down

0 comments on commit 8e3fd61

Please sign in to comment.