MIV-XJTU / SPEED

PyTorch implementation of paper "Sparse Parameterization for Epitomic Dataset Distillation" in NeurIPS 2023.
19 stars 4 forks source link

Nice work! But I got a question about how to change the IPC setting #2

Open HELL-TO-HEAVEN opened 2 months ago

HELL-TO-HEAVEN commented 2 months ago

Nice work! But I got a question about how to change the IPC setting, cuz I didn't find the IPC relevant code. Could you please tell me how to specify IPC? Thanks~

CAOANJIA commented 2 months ago

The calculation of IPC conforms to Eq. (7). We define the default hyperparameters at the end of networks.py, which conform to the corresponding IPC situation. Of course, you can also define your own combination with specific hyperparameters (such as the number of SAETs, the number of non-zero elements in SCMs, etc.), as long as they satisfy Eq. (7).