integrate_so2sat_ref.json

{
    "_description": "armory test integration of pytorch model",
    "adhoc": null,
    "attack": {
        "generate_kwargs": {
            "mask": [
                1,
                1,
                1,
                1,
                0,
                0,
                0,
                0,
                0,
                0,
                0,
                0,
                0,
                0
            ]
        },
        "knowledge": "white",
        "kwargs": {
            "batch_size": 1,
            "eps": 0.031,
            "eps_step": 0.007,
            "max_iter": 20,
            "num_random_init": 1,
            "random_eps": false,
            "targeted": false,
            "verbose": false
        },
        "module": "art.attacks.evasion",
        "name": "ProjectedGradientDescent",
        "use_label": true
    },
    "dataset": {
        "batch_size": 64,
        "eval_split": "validation",
        "framework": "numpy",
        "module": "armory.data.datasets",
        "name": "so2sat"
    },
    "defense": null,
    "metric": {
        "means": true,
        "perturbation": "linf",
        "record_metric_per_sample": false,
        "task": [
            "categorical_accuracy"
        ]
    },
    "model": {
        "fit": false,
        "fit_kwargs": {
            "nb_epochs": 20
        },
        "model_kwargs": {},
        "module": "model_to_integrate.model.so2sat_split",
        "name": "get_art_model",
        "weights_file": "so2sat_split_weights.model",
        "wrapper_kwargs": {
            "clip_values": [
                -1,
                1
            ]
        }
    },
    "scenario": {
        "kwargs": {},
        "module": "armory.scenarios.image_classification",
        "name": "ImageClassificationTask"
    },
    "sysconfig": {
        "docker_image": "twosixarmory/pytorch:0.12.1",
        "external_github_repo": null,
        "gpus": "all",
        "output_dir": null,
        "output_filename": null,
        "use_gpu": false
    }
}