search

MadryLab / robustness

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

add gamma arg for step LR #50

Closed kristian-georgiev closed 4 years ago

kristian-georgiev commented 4 years ago

Small change to allow for different decay rates for a step scheduler from the args.