Hello all,
I am newbie to torch …I am getting this error while passing the new data to the model
"Model "
Sequential(
Linear(in_features=784, out_features=128, bias=True)
ReLU()
Linear(in_features=128, out_features=64, bias=True)
Tanh()
Linear(in_features=64, out_features=10, bias=True)
LogSoftmax(dim=1)
)