EduKTM icon indicating copy to clipboard operation
EduKTM copied to clipboard

Published 20 hours ago • verified publisher icon bigdata-ustc

AKT Bug

Open weizhehuang0827 opened this issue 11 months ago • 0 comments

🐛 Description

https://github.com/bigdata-ustc/EduKTM/blob/main/EduKTM/AKT/AKTNet.py#L273 The mask_fill operation on this line should be replaced with mask_fill_, otherwise it will cause data leakage

What have you tried to solve it?

origin scores.masked_fill(mask == 0, -1e32) solution scores.masked_fill_(mask == 0, -1e32)

weizhehuang0827 avatar Jul 17 '23 06:07 weizhehuang0827