Closed mzweilin closed 1 year ago
This PR tries to improve the progress bar for adversaries.
Epoch n
Attack
Please check all relevant options.
Please describe the tests that you ran to verify your changes. Consider listing any relevant details of your test configuration.
pytest
python -m mart experiment=CIFAR10_CNN_Adv trainer=gpu
python -m mart experiment=CIFAR10_CNN_Adv trainer=ddp trainer.devices=2 model.optimizer.lr=0.2 trainer.max_steps=2925 datamodule.ims_per_batch=256
pre-commit run -a
Make sure you had fun coding 🙃
What does this PR do?
This PR tries to improve the progress bar for adversaries.
Epoch n
withAttack
.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.
pytest
python -m mart experiment=CIFAR10_CNN_Adv trainer=gpu
achieves 70.09% accuracy.python -m mart experiment=CIFAR10_CNN_Adv trainer=ddp trainer.devices=2 model.optimizer.lr=0.2 trainer.max_steps=2925 datamodule.ims_per_batch=256
achieves 69.84% accuracy.Before submitting
pre-commit run -a
command without errorsDid you have fun?
Make sure you had fun coding 🙃