tf-vs-pytorch
tf-vs-pytorch copied to clipboard
typo
Hi, thanks for this great notebook!
At the bottom line, you miswrote return:
reutrn F.log_softmax(x)