docstring for make_madry
@@ -11,6 +11,9 @@ from resnet_tf import ResNetTF
def make_madry_ngpu(nb_classes=10, input_shape=(None, 28, 28, 1), **kwargs):
+ """
+ Create a multi-GPU model similar to Madry et al. (arXiv:1706.06083).
+ """
layers = [Conv2DnGPU(32, (5, 5), (1, 1), "SAME", name='conv1'),
ReLU(name='act1'),
MaxPool((2, 2), (2, 2), "SAME", name='mxpool1'),
