(self, batch)
| 62 | Linear(seq_in_size, config.d_out)) |
| 63 | |
| 64 | def forward(self, batch): |
| 65 | prem_embed = self.embed(batch.premise) |
| 66 | hypo_embed = self.embed(batch.hypothesis) |
| 67 | if self.config.fix_emb: |
| 68 | prem_embed = prem_embed.detach() |
| 69 | hypo_embed = hypo_embed.detach() |
| 70 | if self.config.projection: |
| 71 | prem_embed = self.relu(self.projection(prem_embed)) |
| 72 | hypo_embed = self.relu(self.projection(hypo_embed)) |
| 73 | premise = self.encoder(prem_embed) |
| 74 | hypothesis = self.encoder(hypo_embed) |
| 75 | scores = self.out(torch.cat([premise, hypothesis], 1)) |
| 76 | return scores |
nothing calls this directly
no outgoing calls
no test coverage detected