diff --git a/models/encoder.py b/models/encoder.py index 427eeac..ab61574 100644 --- a/models/encoder.py +++ b/models/encoder.py @@ -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__()