search

TreB1eN / InsightFace_Pytorch

Pytorch0.4.1 codes for InsightFace
MIT License
1.73k stars 423 forks source link

内存问题 #17

Closed shizhenjun closed 5 years ago

shizhenjun commented 5 years ago

https://github.com/TreB1eN/InsightFace_Pytorch/blob/1e0bea082adeca176f5253c38b5903bf95d4db14/utils.py#L34

原始代码会保留梯度，浪费大量内存，建议在这里加上torch.no_grad()，应该可以部分解决问题

TreB1eN commented 5 years ago

已改正 ！