collections defaultdict 使用 模型训练过程中保存每一次的loss输出

from collections import defaultdict

#初始化一个列表型的变量，变量名为“result”
results = defaultdict(list)
for epoch in range(args.epochs):
	...
	results['loss'].append(loss.item())
	results['val_loss'].append(val_loss.item())


np.ravel(list) #去掉嵌套的list