コード例 #1
0
ファイル: transformer.py プロジェクト: PuerJing/NMTGMinor 
 def get_preprocessing_module(self):
     return PrePostProcessing(self.model_dim, 'n', masking=self.masked_layers)
コード例 #2
0
ファイル: transformer.py プロジェクト: PuerJing/NMTGMinor 
 def get_postprocessing_module(self):
     return PrePostProcessing(self.model_dim, 'da', self.residual_dropout, gated_residuals=self.gated_residuals)