Move code into non-executable codeblock.

[jlin27]

In the Neural Network colab, the torch.optim example should be in a non-executable codeblock because it doesn't render an output; otherwise, it confuses the user when it produces an error.

Here's how it looks in the html version: