kulich-d

Results 1 issues of kulich-d

Hi! You have a memory leak during training [here](https://github.com/zyh-uaiaaaa/Erasing-Attention-Consistency/blob/main/src/main.py#L80) It appends because ` print(correct_num) -> ` For solving this problem, I used `.detach()`: ``` loss = loss.detach().cpu() _, predicts =...