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

load models even if you don't have a CPU #127

Open Tal-Golan opened 8 months ago

Tal-Golan commented 8 months ago

Partially addresses issue https://github.com/MadryLab/robustness/issues/15