search

OPTML-Group / Unlearn-Sparse

[NeurIPS23 (Spotlight)] "Model Sparsity Can Simplify Machine Unlearning" by Jinghan Jia*, Jiancheng Liu*, Parikshit Ram, Yuguang Yao, Gaowen Liu, Yang Liu, Pranay Sharma, Sijia Liu
MIT License
62 stars 7 forks source link

Mistake in code? #6

Closed rohan1561 closed 8 months ago

rohan1561 commented 9 months ago

Hello, is there a problem in the line 259 of dataset.py? It appears as if indexes_to_replace has been considered to be an int and is supposed to be a list.. was the or clause supposed to be added to line 256 with num_indexes_to_replace? Thanks!

jinghanjia commented 8 months ago

Thank you for highlighting this issue. We have corrected the mistake, which occurred during the code cleanup process. Your assistance is greatly appreciated.