http://pytorch.org/docs/master/nn.html#torch.nn.LogSoftmax 92
LogSoftmax takes 1 argument, it sounds like you’re calling it with 2.
If you’re getting “RuntimeError: Assertion 'cur_target >=0 && cur_target < n_classes failed.” --> I don’t know what’s in your dataset, but have you checked the number of classes you have?