Skip to content
This repository has been archived by the owner on Jul 18, 2024. It is now read-only.

version of pytorch #1

Open
miaoxiaodaiblack opened this issue Mar 6, 2021 · 3 comments
Open

version of pytorch #1

miaoxiaodaiblack opened this issue Mar 6, 2021 · 3 comments

Comments

@miaoxiaodaiblack
Copy link

Hi!
Your work is so interesting! Thank you!
But is there the code written by pytorch?

@huanzhang12
Copy link
Collaborator

@miaoxiaodaiblack Thank you so much for your interests in our work!

In our CROWN-IBP repository we provide a PyTorch implementation of CROWN on simple convolutional neural networks . See instructions here.

Moreover, we developed the auto_LiRPA library which implements CROWN in PyTorch for general neural network architectures. We support CROWN based certified training and verification on fairly complicated networks like ResNeXt, DenseNet, LSTM and Transformer. The library is user friendly and easy to use: you only need to define a forward() function in PyTorch and you can obtain bounds automatically without any manual derivations. We provide many examples for our library and you can also checkout our Colab Demo for a quick start.

Both libraries are very efficient and run on (multi-)GPUs with PyTorch, and can handle relatively large networks. Feel free to let me know if you need any additional help!

Thanks,
Huan Zhang

@miaoxiaodaiblack
Copy link
Author

miaoxiaodaiblack commented Mar 22, 2021 via email

@miaoxiaodaiblack
Copy link
Author

Thank you for your reply!

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants