我需要自定义metric,因此ms2.0要求一定要设置eval_network。在第一个epoch,显存占用是13GB,经过一次eval之后,显存占用为19GB,之后一直保持19GB。
并且还发现,训练的loss在第一个epoch当中不会下降,在第二个epoch刚开始时骤降,感觉训练过程中并没有更新参数。
model = Model(net, optimizer=optimizer,eval_network=net,eval_indexes =[0,1,2],metrics={"PRF":PRF(dev_ner)})
训练过程中loss是不怎么变的
****************************************************解答*****************************************************
目前MindSpore已提供函数式编程,可以不用model封装
https://mindspore.cn/tutorials/zh-CN/r2.1/beginner/autograd.html