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 decoder.py

main
BBing 2 years ago
parent
9e977a6bd0
commit
b613fc8f91
1 changed files with 1 additions and 0 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -0
      models/decoder.py

+ 1
- 0
models/decoder.py View File

@@ -1,6 +1,7 @@
import mindspore.nn as nn
import mindspore.ops.operations as ops


class DecoderLayer(nn.Cell):
def __init__(self, self_attention, cross_attention, d_model, d_ff=None,
dropout=0.1, activation="relu"):


Write Preview
Loading…
Cancel
Save