Hi, thanks for this great notebook! At the bottom line, you miswrote `return`: `reutrn F.log_softmax(x)`
Hi, thanks for this great notebook!
At the bottom line, you miswrote
return:reutrn F.log_softmax(x)