Commit ab5c881d authored by Alessia Marcolini's avatar Alessia Marcolini
Browse files

REMOVE SOFTMAX AFTER THE LAST LINEAR LAYER, CROSSENTROPY LOSS ALREADY INCLUDES SOFTMAX

parent 30c1d40b
......@@ -95,7 +95,7 @@ class CiompiDO(nn.Module):
nn.Dropout(self.dropout/2),
nn.Linear(50, self.n_classes),
nn.Softmax(1)
#nn.Softmax(1)
)
def forward(self, x): #x = N_batch x N_ch * s x s x s
......
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