From 4dc14a24893f6bbbcdfd1bf9d363e414580af13e Mon Sep 17 00:00:00 2001 From: Yuchen Fang Date: Thu, 28 Jan 2021 00:41:22 +0800 Subject: [PATCH] minor --- examples/trade/model/opd.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/examples/trade/model/opd.py b/examples/trade/model/opd.py index bf426f218..6a25d0f35 100644 --- a/examples/trade/model/opd.py +++ b/examples/trade/model/opd.py @@ -20,15 +20,10 @@ class OPD_Extractor(nn.Module): self.rnn2 = nn.GRU(64, hidden_size, batch_first=True) self.dnn = nn.Sequential(nn.Linear(2, 64), nn.ReLU(),) self.cnn = nn.Sequential(nn.Conv1d(self.cnn_shape[1], 3, 3), nn.ReLU(),) - self.raw_fc = nn.Sequential( - nn.Linear((self.cnn_shape[0] - 2) * 3, 64), nn.ReLU(), - ) + self.raw_fc = nn.Sequential(nn.Linear((self.cnn_shape[0] - 2) * 3, 64), nn.ReLU(),) self.fc = nn.Sequential( - nn.Linear(hidden_size * 2, hidden_size), - nn.ReLU(), - nn.Linear(hidden_size, 32), - nn.ReLU(), + nn.Linear(hidden_size * 2, hidden_size), nn.ReLU(), nn.Linear(hidden_size, 32), nn.ReLU(), ) def forward(self, inp):