i reduced the learning rate to 10^-5 and the outputs are now different. Thank you. but the problem is that in the o/p, the highest probability always occurs at one index only, say 2. So the predicted class obtained is 2 for all data. I’d run several epochs but am getting same class value. Kindly suggest what should i do ?
validset data   : tensor([[0.1263, 0.9880, 0.8812, 0.5894]])
validset output         : tensor([[0.0481, 0.0847, 0.1271]], grad_fn=<AddmmBackward>)
target output is:        tensor([1])
label output is:         2
label target is:         1


validset data   : tensor([[0.2822, 0.1722, 0.3539, 0.0999]])
validset output         : tensor([[0.0201, 0.0679, 0.1299]], grad_fn=<AddmmBackward>)
target output is:        tensor([1])
label output is:         2
label target is:         1


validset data   : tensor([[0.1996, 0.1763, 0.7774, 0.9449]])
validset output         : tensor([[0.0392, 0.0913, 0.1218]], grad_fn=<AddmmBackward>)
target output is:        tensor([2])
label output is:         2
label target is:         2


validset data   : tensor([[0.3247, 0.0582, 0.7237, 0.0739]])
validset output         : tensor([[0.0199, 0.0653, 0.1324]], grad_fn=<AddmmBackward>)
target output is:        tensor([0])
label output is:         2
label target is:         0


validset data   : tensor([[0.4353, 0.9065, 0.1725, 0.4158]])
validset output         : tensor([[0.0432, 0.0784, 0.1270]], grad_fn=<AddmmBackward>)
target output is:        tensor([2])
label output is:         2
label target is:         2


validset data   : tensor([[0.9252, 0.5795, 0.1595, 0.7732]])
validset output         : tensor([[0.0479, 0.0946, 0.1213]], grad_fn=<AddmmBackward>)
target output is:        tensor([2])
label output is:         2
label target is:         2

output: 2        target: 0
output: 2        target: 0
output: 2        target: 1
output: 2        target: 0
output: 2        target: 2
output: 2        target: 1
output: 2        target: 1
output: 2        target: 1
output: 2        target: 0
output: 2        target: 1
output: 2        target: 2
output: 2        target: 1
output: 2        target: 1
output: 2        target: 0
output: 2        target: 2
output: 2        target: 0
output: 2        target: 2
output: 2        target: 1