Commit 9a624e84 authored by Marco Cristoforetti's avatar Marco Cristoforetti
Browse files

Merge branch 'master' of gitlab.fbk.eu:mcristofo/DST

parents a2b775a6 bb8579ed
data_path = '/home/marco/projects/projects_data/DST/data/'
data_path = '/storage/DSIP/DST/data/'
This source diff could not be displayed because it is too large. You can view the blob instead.
This source diff could not be displayed because it is too large. You can view the blob instead.
This source diff could not be displayed because it is too large. You can view the blob instead.
......@@ -199,9 +199,9 @@ class DSTnet(nn.Module):
self.linear_o_3 = nn.Linear(self.nhidden_o, self.nhidden_o // 2)
self.linear_o_4_c = nn.Linear(self.nhidden_o // 2, self.after * 4)
self.linear_o_4_r = nn.Linear(4, 2)
# self.linear_o_4b_r = nn.Linear(2, 2)
self.linear_o_5_r = nn.Linear(2, 1)
self.linear_o_4_r = nn.Linear(4, 8)
self.linear_o_4b_r = nn.Linear(8, 4)
self.linear_o_5_r = nn.Linear(4, 1)
def init_hidden(self, batch_size):
......@@ -225,8 +225,8 @@ class DSTnet(nn.Module):
x = F.relu(self.linear_o_3(x))
x = F.dropout(x, 0.2, training=self.training)
x1 = self.linear_o_4_c(x)
x2 = self.linear_o_4_r(x1.view(-1, 4))
# x2 = self.linear_o_4b_r(x2)
x2 = F.relu(self.linear_o_4_r(x1.view(-1, 4)))
x2 = F.relu(self.linear_o_4b_r(x2))
x2 = self.linear_o_5_r(x2)
x2 = x2.reshape(x0.size(0), self.after)
x1 = x1.reshape(x0.size(0) * self.after, 4)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment