SACN icon indicating copy to clipboard operation
SACN copied to clipboard

Published 20 hours ago • verified publisher icon JD-AI-Research-Silicon-Valley

when training the SACN on WN18RR, CUDA is our of memory(The batch size is 64)

Open learntolearn11 opened this issue 4 years ago • 2 comments

learntolearn11 avatar Jul 09 '20 12:07 learntolearn11

try torch.cuda.empty_cache() after every batch may help free about 2GB memory(in my experiment).

maqy1995 avatar Sep 22 '20 07:09 maqy1995

@maqy1995 Thanks for your help to response this question.

chaoshangcs avatar Sep 22 '20 22:09 chaoshangcs