search

OPTML-Group / Unlearn-Saliency

[ICLR24 (Spotlight)] "SalUn: Empowering Machine Unlearning via Gradient-based Weight Saliency in Both Image Classification and Generation" by Chongyu Fan*, Jiancheng Liu*, Yihua Zhang, Eric Wong, Dennis Wei, Sijia Liu
https://www.optml-group.com/posts/salun_iclr24
MIT License
90 stars 12 forks source link

Regarding GPU Information #5

Closed thakral-kartik closed 6 months ago

thakral-kartik commented 6 months ago

Hi,

I really liked your work!!! Could you clarify the GPU requirements needed to train the models?

Thanks, Kartik

a-F1 commented 6 months ago

Thank you for your interest in our work! Our experiments are conducted on the NVIDIA RTX A6000.