Solved by KFrank in post #4 
              
              
                Hello John! 

It depends on what you are trying to do, and what your loss function is. 
It sounds as if you might be working on a twenty-class classification 
problem.  In such a case, you would typically use CrossEntropyLoss 
as your loss function.  Your model would output directly the results of 
…