This commit is contained in:
robcaulk 2023-05-06 17:40:04 +00:00
parent 3bbb7e38ea
commit 36e1e58dad

View File

@ -39,7 +39,7 @@ class PyTorchTransformerModel(nn.Module):
# the pseudo decoding FC
self.output_net = nn.Sequential(
nn.Linear(hidden_dim * time_window, int(hidden_dim)),
nn.Linear(self.dim_val * time_window, int(hidden_dim)),
nn.ReLU(),
nn.Dropout(dropout_percent),
nn.Linear(int(hidden_dim), int(hidden_dim / 2)),