This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
论坛
实训
竞赛
大数据
AI开发
Register
Sign In
limingjuan
/
Prediction-MindSpore
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
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
6877a32962
commit
d07f92920e
1 changed files
with
2 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+2
-1
models/decoder.py
+ 2
- 1
models/decoder.py
View File
@@ -1,7 +1,8 @@
import torch
import torch.nn as nn
import torch.nn.functional as F
1
class DecoderLayer(nn.Module):
def __init__(self, self_attention, cross_attention, d_model, d_ff=None,
dropout=0.1, activation="relu"):
Write
Preview
Loading…
Cancel
Save