folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix : gpu loss plotting

Browse Source
This commit is contained in:
mathayay
2023-11-29 16:31:28 +01:00
committed by Nicola Demo
parent 42cf7fcaf1
commit 4639374961
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pina/plotter.py
View File

@@ -256,7 +256,7 @@ class Plotter:
) )
loss = trainer_metrics[metric] loss = trainer_metrics[metric]
epochs = range(len(loss)) epochs = range(len(loss))
plt.plot(epochs, loss, **kwargs) plt.plot(epochs, loss.cpu(), **kwargs)
# plotting # plotting
plt.xlabel('epoch') plt.xlabel('epoch')