(self, x)
| 24 | self.drop = nn.Dropout(drop_rate) |
| 25 | |
| 26 | def forward(self, x): |
| 27 | x = self.fc1(x) |
| 28 | x = self.act(x) |
| 29 | if self.drop_rate > 0.0: |
| 30 | x = self.drop(x) |
| 31 | x = self.fc2(x) |
| 32 | if self.drop_rate > 0.0: |
| 33 | x = self.drop(x) |
| 34 | return x |
| 35 | |
| 36 | |
| 37 | class Permute(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected