BACKPROP
VISUALIZATION
BUILT FOR DL-PYTORCH WEEK #4
y_ = model(x)
loss = criterion(y_, y)
loss.backward()
optimizer.step()
y_ = model(x)
loss = criterion(y_, y)
loss.backward()
optimizer.step()

Backprop Visualization