... BatchNorm1d(config.fc1_dim), nn.Dropout(p=config.dropout), nn.Linear(config.fc1_dim, config.fc2_dim) ) def forward(self, batch): prem_embed = self.embed ...
確定! 回上一頁