Log softmax

[roserustowicz]

• Convert all torch.log(nn.2DSoftmax()) cases to nn.LogSoftmax(dim=1)
• A few other small fixes throughout