Commit 739c1dbf authored by Alexis Brenon's avatar Alexis Brenon
Browse files

Completly remove non-parametric closing layers

These layers make the network not learn anything...
parent 61886639
......@@ -44,8 +44,6 @@ function module.create_network(args)
-- Output layer
net:add(nn.Linear(nelements, torch.Tensor(args.output_size):prod()))
-- net:add(args.nl_layer(true))
-- net:add(nn.LogSoftMax())
net:add(nn.View(unpack(args.output_size)))
return net
end
......
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