Update decoder.py · b613fc8f91 - Prediction-MindSpore - 开源协同云脑生态支撑系统

Update decoder.py

main

BBing 2 years ago

parent

9e977a6bd0

commit

b613fc8f91

1 changed files with 1 additions and 0 deletions

Unified View

							
								+ 1
								
								- 0
							
models/decoder.pyView File
								
																@@ -1,6 +1,7 @@
															
																import mindspore.nn as nn
																
																import mindspore.nn as nn
															
																import mindspore.ops.operations as ops
																
																import mindspore.ops.operations as ops
															
																class DecoderLayer(nn.Cell):
																
																class DecoderLayer(nn.Cell):
															
																    def __init__(self, self_attention, cross_attention, d_model, d_ff=None,
																
																    def __init__(self, self_attention, cross_attention, d_model, d_ff=None,
															
																                 dropout=0.1, activation="relu"):
																
																                 dropout=0.1, activation="relu"):