search

melihcatal / advsecurenet

Machine Learning Security Library
https://melihcatal.github.io/advsecurenet/
MIT License
4 stars 3 forks source link

CLI model config should support IMAGENET1K_V2 weights for the standard pretrained models #138

Closed melihcatal closed 6 months ago

melihcatal commented 6 months ago

The package supports IMAGENET1K_V2 weights through API, but users should also be able to select the weights on the CLI.