IntelLabs / MART

Modular Adversarial Robustness Toolkit
BSD 3-Clause "New" or "Revised" License
16 stars 0 forks source link

Add a YAML config file generator #224

Closed mzweilin closed 9 months ago

mzweilin commented 9 months ago

What does this PR do?

This PR adds a YAML config file generator.

We can generate YAML configs for running MART attacks in external projects. For example,

python -m mart.generate_config \
--export_node=attack \
+attack=classification_fgsm_linf \
attack.eps=0.5 \
> attack_config.yaml

Type of change

Please check all relevant options.

Testing

Please describe the tests that you ran to verify your changes. Consider listing any relevant details of your test configuration.

Before submitting

Did you have fun?

Make sure you had fun coding 🙃