Hi @Isaac_Kargar, Thank you for your answer.
I know, but I got a problem in the loss function:
my code:
loss = criterion(y_pred, y_real.to(device))
my criterion is BCE
ValueError: Target size (torch.Size([10003])) must be the same as input size (torch.Size([10003, 1]))
Thats why I started to consider about the output from the NN.
Do I need to reshape the output before using the criterion function?
Thank you