Plot all loss values for wgan moedel

1

I asked for how I can plot the model losses with number of iterations
that my train_losses.append([g_loss.item(), d_loss.item()-gp_loss.item(), gp_loss.item()])
neither only plottting generator loss nor only disciminator loss
hope plot the loss values for all model loss .
is that correct plt.plot(np.array(train_losses), ‘r’)?

2

Yes, your approach might work, as matplotlib should be smart enough to be able to plot a list containing multiple values in each element:

a = []
for _ in range(10):
    a.append([torch.randn(1).item(), torch.randn(1).item(), torch.randn(1).item()])

plt.plot(a)
3

Thanks, I realized and Solved my issue.