1、保存所有的loss

all_loss=[]
all_loss.append(loss)
file=open('data.txt','w')
file.write(all_loss)
file.close()

2、画图

list = []
with open('data.txt', 'r') as f:for line in f.readlines():arr = line.split(',')
print(arr.__len__())
arr[0]=arr[0][1:]
arr[-1]=arr[-1][0:-2]
print(arr[0])
print(arr[-1])
for i in range(len(arr)):arr[i]=float(arr[i])
import matplotlib.pyplot as plt
x=[]
for i in range(len(arr)):x.append(i*100)
plt.plot(x,arr,label="ssd-loss")
plt.legend()
plt.show()