add missing kwargs
@@ -552,7 +552,7 @@ class BasicIterativeMethod(ProjectedGradientDescent):
class MadryEtAl(ProjectedGradientDescent):
- def __init__(self, model, sess=None, dtypestr='float32'):
+ def __init__(self, model, sess=None, dtypestr='float32', **kwargs):
super(MadryEtAl, self).__init__(model, sess=sess,
dtypestr=dtypestr,
default_rand_init=True,
