Browse Source

Update encoder.py

main
BBing 2 years ago
parent
commit
c7bd34250a
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      models/encoder.py

+ 1
- 0
models/encoder.py View File

@@ -5,6 +5,7 @@ import mindspore.tensor as Tensor
import mindspore.common.dtype as mstype
import mindspore.common.dtype as mstype


class ConvLayer(nn.Cell):
def __init__(self, c_in):
super(ConvLayer, self).__init__()


Loading…
Cancel
Save