search

Zinoex / bound_propagation

Linear and interval bound propagation in Pytorch with easy-to-use API and GPU support.
GNU General Public License v3.0
7 stars 3 forks source link

Conv net support? #8

Open vihari opened 1 year ago

vihari commented 1 year ago

Hi, Thanks a lot for this very useful repository. Cleaning up IBP interface is much needed. I am just wondering if your code supports conv nets. Thanks!

Zinoex commented 1 year ago

Hi Vihari,

Could you please clarify what you mean by cleaning up the IBP interface? More specifically what would you like to see improved, and how do you think the interface should be changed?

The code does not currently support convolutional neural nets, and I have no plans of adding them as they are outside the scope of my research. Feel free to take a stab at the problem yourself and open a pull request. If you need something that is working right now, I can recommend auto_LiRPA.

Best regards, Frederik