MadryLab / robustness

A library for experimenting with, training and evaluating neural networks, with a focus on adversarial robustness.
MIT License
903 stars 181 forks source link

Architecture Types #120

Open pstefanou12 opened 1 year ago

pstefanou12 commented 1 year ago

The check_and_fill_args function does not catch when you pass in an architecture that the package doesn't support or you spell it incorrectly. You don't get a failure until you call the make_and_restore_model function. Not a big deal, but this should fail earlier.