Commit a67fb39c authored by Marco Cristoforetti's avatar Marco Cristoforetti
Browse files

nothing

parent a65286f9
data_path = '/home/marco/projects/projects_data/DST/data/'
data_path = '/storage/DSIP/DST/data/'
data_path = '/home/marco/projects/projects_data/DST/data/'
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
......@@ -2,7 +2,7 @@
"cells": [
{
"cell_type": "code",
"execution_count": null,
"execution_count": 1,
"metadata": {},
"outputs": [],
"source": [
......@@ -25,7 +25,7 @@
},
{
"cell_type": "code",
"execution_count": null,
"execution_count": 2,
"metadata": {},
"outputs": [],
"source": [
......@@ -37,7 +37,7 @@
},
{
"cell_type": "code",
"execution_count": null,
"execution_count": 3,
"metadata": {},
"outputs": [],
"source": [
......
......@@ -197,7 +197,7 @@ class DSTnet(nn.Module):
self.linear_o_3 = nn.Linear(self.nhidden_o, self.nhidden_o // 2)
self.linear_o_4 = nn.Linear(self.nhidden_o // 2, self.after)
self.linear_o_4_c = nn.Linear(self.nhidden_o // 2, self.after*4)
self.linear_o_4_c = nn.Linear(1, 4)
def init_hidden(self, batch_size):
......@@ -221,8 +221,9 @@ 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(x)
x2 = self.linear_o_4_c(x)
x2 = x2.reshape(x0.size(0) * self.after, 4)
return x1, x2
......
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