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

Tal-Golan wants to merge 1 commit intoMadryLab:masterMadryLab/robustness:masterfrom

brainsandmachines:masterbrainsandmachines/robustness:masterCopy head branch name to clipboard