Example #1
0
 def __init__(self):
     super(UnfoldNetSame, self).__init__()
     self.unfold = nn.Unfold(ksizes=[1, 2, 2, 1],
                             strides=[1, 1, 1, 1],
                             rates=[1, 1, 1, 1],
                             padding='SAME')
Example #2
0
 def __init__(self):
     super(UnfoldNetValid, self).__init__()
     self.unfold = nn.Unfold(ksizes=[1, 2, 2, 1],
                             strides=[1, 1, 1, 1],
                             rates=[1, 1, 1, 1],
                             padding='VALID')