Open programehr opened 1 year ago
Hi,
I realized that your defense code does not work with networks without batchnorm layers. Is it intentional/expected or is your algorithm supposed to handle the said kind of models as well?
Regards, Mehrin
https://github.com/rkteddy/channel-Lipschitzness-based-pruning/blob/5209c837abfe1c40b0ce3d2971f9f83c172955fd/defense.py#L12
Hi,
I realized that your defense code does not work with networks without batchnorm layers. Is it intentional/expected or is your algorithm supposed to handle the said kind of models as well?
Regards, Mehrin
https://github.com/rkteddy/channel-Lipschitzness-based-pruning/blob/5209c837abfe1c40b0ce3d2971f9f83c172955fd/defense.py#L12