Browse Source

add A2Attention

pull/1/head
Huyf9 2 years ago
parent
commit
ba099c2f7a
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model/attention/A2Attention.py

+ 1
- 1
model/attention/A2Attention.py View File

@@ -1,7 +1,7 @@
import mindspore as ms
from mindspore import nn
from mindspore.common.initializer import initializer, HeNormal, Normal

class DoubleAttention(nn.Cell):
def __init__(self, in_channels, c_m, c_n, reconstruct=True):


Loading…
Cancel
Save