part 1: create feedforward network, print output
part 2: print prediciton, loss, create optimizer, calculate gradients, and run optimizer
(Edit: note, this hsould be log_softmax, rather than softmax 
)
3 Likes