pytorch retain_graph=True

发表于 2018-08-Mon | 阅读次数:

pytorch 在backward的时候会删除计算图，
所以在backward的时候要加入参数 loss.backward(retain_graph=True)

请作者喝一杯咖啡☕️