【发布时间】:2018-10-15 23:41:17
【问题描述】:
我找到了一行代码:
out_a, out_p, out_n = model(data_a), model(data_p), model(data_n)
在:https://github.com/liorshk/facenet_pytorch/blob/master/train_triplet.py
你可以看到在“backward()”之前多次调用了“forward()”,问题是:
1,下一次调用会覆盖之前的计算图吗?
2,在我的测试中,GPU消耗会相应增加,所以即使发生override,也会出现GPUu内存泄漏,如何解决这个问题?
【问题讨论】:
标签: pytorch