Changes

DPS921/PyTorch: Convolutional Neural Networks

6 bytes added, 14:15, 30 November 2020

→‎Parallelization Methods

<code>model = ToyModel()</code> <code>loss_fn = nn.MSELoss()</code> <code>optimizer = optim.SGD(model.parameters(), lr=0.001)</code>

<code>optimizer.zero_grad()</code>

56

edits