save plot

This commit is contained in:
ltcptgeneral 2024-03-07 00:54:47 -08:00
parent 6cbffcdec1
commit c121415e31

View File

@ -58,6 +58,4 @@ for i in range(epochs):
plt.plot(losses, color="green", label="Training Loss")
plt.legend(loc = 'upper left')
plt.show()
plt.savefig("plot.png")