Необходимо построить график объекта [0, 0] тензора pytorch 2d против значения потерь - PullRequest
Новая
       12

Необходимо построить график объекта [0, 0] тензора pytorch 2d против значения потерь

0 голосов
/ 27 мая 2020

Мне нужен график вроде этого:

/\w/ with a tangent in the first upslope in the w

Вот мой код 

new_weights_tensor = w.clone()

weights_x = [new_weights_tensor[0, 0].item()]
loss_y = [float(loss.detach().numpy())]

for i in range(0, 10):
    new_weights_tensor = new_weights_tensor.clone()
    new_weights_tensor[0, 0] = (new_weights_tensor[0, 0] + 0.01)
    pred = model(inputs, new_weights_tensor)
    new_loss = mse(pred, targets)

    weights_x.append(new_weights_tensor[0, 0].item())
    loss_y.append(new_loss.item())

# print(weights_x)
# print(loss_y)

plt.plot(weights_x, loss_y)
plt.show()

Но вместо этого я получаю это:

Downward sloping line

Это ссылка на учебник, по которому я следую https://jovian.ml/aakashns/02-linear-regression

...