# Build a fully connected layer.
# Build MSE loss function and optimizer.
# Optimization method using RMSprop
optimizer = torch.optim.RMSProp(linear.parameters(), lr=1.0, rho=0.9, eps=1e-06, weight_decay=0)
loss = criterion(pred, y)
print('loss:', loss.item())