Armory has several baseline models available for use in evaluations. All of these models return an ART wrapped classifier for use with ART attacks and defenses.
Pretrained weights can be loaded in to the baseline models or custom models. This is
achieved by specifying the name in the
weights_file field of a model's config.
When the model is loaded it will first try to load the file from the armory
saved_model_dir. This enables you to place your own custom weights in that directory
for loading. If the weights file is not found it'll then try to download the file from
our S3 bucket. Files that are available in the armory S3 bucket are listed in the table
weights_file is not found locally or in the S3 bucket an error will be
The model files can be found in armory/baseline_models/keras.
The model files can be found in armory/baseline_models/pytorch
The model file can be found in armory/baseline_models/tf_graph. The weights for this model are downloaded from the link listed below.
|Model||TF Weights URL|
Preprocessing functions have been moved inside each model's forward pass. This is to allow each model to receive as input the canonicalized form of a dataset.