Plot all loss values for wgan moedel

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’)?

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()])


Thanks, I realized and Solved my issue.