import torch
from torch import nn
from torch.utils.tensorboard import SummaryWriter


torch.manual_seed(7)
features = torch.arange(-5, 5, 0.1).view(-1, 1)
targets = -5 * features + 0.1 * torch.randn(features.size())
model = nn.Linear(1, 1)
loss_fn = nn.MSELoss()
optimizer = torch.optim.SGD(model.parameters(), lr=0.01)
writer = SummaryWriter("runs/linear-regression")