pytorch mlp add explicit annotation to fix mypy error
This commit is contained in:
parent
6b204c97ed
commit
26738370c7
@ -48,7 +48,7 @@ class PyTorchMLPModel(nn.Module):
|
||||
self.dropout = nn.Dropout(p=dropout_percent)
|
||||
|
||||
def forward(self, x: List[torch.Tensor]) -> torch.Tensor:
|
||||
x = x[0]
|
||||
x: torch.Tensor = x[0]
|
||||
x = self.relu(self.input_layer(x))
|
||||
x = self.dropout(x)
|
||||
x = self.blocks(x)
|
||||
|
Loading…
Reference in New Issue
Block a user