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

Update encoder.py

main
BBing 2 years ago
parent
4ba099e51a
commit
78dd836555
1 changed files with 1 additions and 1 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      models/encoder.py

+ 1
- 1
models/encoder.py View File

@@ -93,4 +93,4 @@ class EncoderStack(nn.Cell):
attns.append(attn)
x_stack = ops.Concat(2)(x_stack)

return x_stack, attns
return x_stack, attns

Write Preview
Loading…
Cancel
Save