limingjuan
/
Prediction-MindSpore
Not watched
1
0
Fork 0
Code
Releases 0 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

add A2Attention

pull/1/head
Huyf9 2 years ago
parent
5bc0c8ced1
commit
ba099c2f7a
1 changed files with 1 additions and 1 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  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):


Write Preview
Loading…
Cancel
Save